Skip to content

Latest commit

 

History

History
19 lines (13 loc) · 556 Bytes

Readme.md

File metadata and controls

19 lines (13 loc) · 556 Bytes

Cats vs Dogs in Pytorch

Getting my hands on pytorch

Note that the network implemented here is not exactly AlexNet.

Dependencies

  • Pytorch
  • Tensorboard
  • opencv
  • tqdm

Train

Run main.py, you may want to look at main.py though to specify the dataset location for instance.

To visualize training progress, execute tensorboard --logdir=tensorboard_logs on your terminal and open the given url on your browser.

Dataset can be downloaded from https://www.microsoft.com/en-us/download/details.aspx?id=54765&WT.mc_id=rss_alldownloads_devresources