kuanhungchen / CIFAR10-TODO.pytorch

(HOMEWORK)

CIFAR10-TODO.pytorch

Requirements

• Python 3.6
• PyTorch 0.4.1

Usage

1. Train

   $ python train.py -d=./data -c=./checkpoints
   

2. Evaluate

   $ python eval.py ./checkpoint/model-100.pth -d=./data
   

3. Infer

   $ python infer.py ./images/dog.jpg -c=./checkpoints/model-100.pth