This project borrows ideas from PyTorch-Template, aimming to provide a easy and extensible template for my future pytorch projects.
├── datasets
(wrapper for datasets)
├── models
(network descriptions)
├── options
(hyper-parameters parser for command line inputs)
├── trainers
(define how model's forward / backward and logs)
├── utils
(toolbox for drawing and scheduling)
├── cifar.py
(main .py)
├── main.sh
(setup for the project)
└── work
(default folder to store logs/models)
- Train ResNet-20 (pre-activation) on CIFAR10
python cifar.py \ /home/zzh/dataset/cifar10 \ --arch resnet20 \ --dataset cifar10
- Train ResNet-18 on ImageNet
sh main.sh