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]
trainer.fast_dev_run True
debug,add command line 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.