/DogsVsCats-ResNet18

use torchvision models ResNet18 to implement Kaggle's dogs vs. cats task

Primary LanguagePython

DogsVsCats-ResNet18

use torchvision models ResNet18 to implement Kaggle's dogs vs. cats task

Hardware

Intel 6400, 16G RAM, GTX960-2GB

OS and DL Tools

Ubuntu 16.04, Anaconda, PyTorch 1.1.0, Python3.6

Run

  1. set data and checkpoint folders directory in all .py files
  2. RUN python train.py
  3. RUN test.py