This is an implementaiton of AlexNet, as introduced in the paper "ImageNet Classification with Deep Convolutional Neural Networks" by Alex Krizhevsky et al.

• python>=3.9.12

• torch==1.13.1

• torchvision==0.14.1

You can install required packages by:

pip3 install -r requirements.txt

python3 train.py

Specify TRAIN_IMG_DIR in the script before training.

If you find a bug, create a GitHub issue, or even better, submit a pull request. Similarly, if you have questions, simply post them as GitHub issues.