This is a implementation of Alexnet. The network is trained and tested on cat vs dog dataset of Kaggle. It is built based on PyTorch.
The pretrained model can be downloaded from Pytorch website. And the pretrained model should be put into the data folder.
If you want to training the alxenet by yourself, the cat and dog dataset should be downloaded from kaggle.com and put into the data folder.