A collection of deep learning paper implementations in PyTorch, trained and evaluated on standard benchmarks.
| # | Paper | Dataset | Directory |
|---|---|---|---|
| 1 | AlexNet — ImageNet Classification with Deep Convolutional Neural Networks (Krizhevsky et al., 2012) | CIFAR-10 | AlexNet/ |
PyTorch implementation of AlexNet adapted for CIFAR-10.
- Python 3.8+
- PyTorch
- TorchVision
cd AlexNet
pip install -r requirements.txtpython train.py| Argument | Description | Default |
|---|---|---|
--data-dir |
Directory to store CIFAR-10 data | ./data |
--epochs |
Number of training epochs | 5 |
--batch-size |
Batch size for training | 64 |
--lr |
Learning rate | 0.01 |
--momentum |
Momentum for SGD | 0.9 |
--weight-decay |
Weight decay for SGD | 0.0005 |
--save-path |
Path to save trained model weights | alexnet.pth |
Example:
python train.py --epochs 10 --batch-size 128 --lr 0.005| File | Description |
|---|---|
train.py |
AlexNet model definition and training loop |
alexnet.ipynb |
Jupyter notebook for interactive exploration |
requirements.txt |
Python dependencies |
NIPS-2012-imagenet-classification-with-deep-convolutional-neural-networks-Paper.pdf |
Original paper |
Each paper lives in its own subdirectory with a self-contained train.py, requirements.txt, and the original PDF.