aiterahai / alexnet-pytorch

PyTorch Implementation of AlexNet

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.