/alexnet-pytorch

PyTorch Implementation of AlexNet

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

alexnet-pytorch


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

Prerequisites


  • python>=3.9.12

  • torch==1.13.1

  • torchvision==0.14.1

You can install required packages by:

pip3 install -r requirements.txt

Training

python3 train.py

Specify TRAIN_IMG_DIR in the script before training.

Contributing

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.