marsggbo / torchline

Easy to use Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

Easy to use Pytorch

License:MIT License


Languages

Language:Python 99.8%Language:Shell 0.2%