/CIFAR10-TODO.pytorch

Primary LanguagePython

(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