Northengard/dl_pipe

Deep learning pipe

NN Training pipeline for dfcd task.

Prepare to use:

• git clone https://github.com/Northengard/dl_pipe
• cd dl_pipe
• pip install requirements.txt

Usage:

• create your config file for training. See example in configs.
• insert your own model and data_set if necessary
• python train.py --config <Your_config_file.yaml>

You can follow an example of "ForwardRegression model" and DFC dataset

Pretrained weights are available. This snapshot has ~86 accuracy on validation set