/torchline

Easy to use Pytorch

Primary LanguagePythonMIT LicenseMIT

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