torchline v0.3.0.4

Easy to use Pytorch

Only one configure file is enough!

You can change anything you want just in only one configure file.

Dependences

• Python>=3.6
• Pytorch>=1.3.1
• torchvision>=0.4.0,<0.5.0
• yacs==0.1.6
• pytorch-lightning<=0.7.6

Install

• Before you install torchline, please make sure you have installed the above libraries.
• You can use torchline both in Linux and Windows.

pip install torchline

Run demo

train model with GPU0 and GPU 1

cd projects/cifar10_demo
python main.py --config_file cifar10.yaml trainer.gpus [0,1]

debug，add command line trainer.fast_dev_run True

cd projects/cifar10_demo
python main.py --config_file cifar10.yaml trainer.gpus [0] trainer.fast_dev_run True

CIFAR demo uses ResNet50，which is trained for 72 epochs and achieved the best result (94.39% validation accuracy) at the epoch 54.

Thanks

• AutoML
• pytorch-lightning