Modified Implementation of AlexNet

The AlexNet implementation in this repository is based on
the implementation by kratzert.

The AlexNet architecture in this implementation deviates from the
original implementation, so as to use the CIFAR-10
dataset, which are smaller in dimensions compared to the ImageNet images.

Beside the comments in the code itself, I also wrote an article which you can find here with further explanations on AlexNet.

Content

• alexnet.py: Class with the graph definition of the AlexNet.
• AlexNet Train.ipynb: Notebook to train and test the implementation of AlexNet.

If you are having trouble viewing the notebook file, click here.