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