Requirements Python 3.6 PyTorch 0.4.1 Usage Train $ python train.py -d=./data/processed -c=./checkpoints Evaluate $ python eval.py ./checkpoints/model-100.pth -d=./data/processed Infer $ python infer.py ./images/dog.jpg -c=./checkpoints/model-100.pth CV_final