This project implements a custom ResNet model optimized to stay under 5 million parameters while maintaining high classification accuracy on the CIFAR-10 dataset. It was submitted as part of a Kaggle competition.
- Lightweight ResNet architecture
- Parameter count constrained below 5M
- Achieved strong test accuracy on CIFAR-10
- CIFAR-10 (10-class image classification)
- Python
- PyTorch
- Torchvision
pip install torch torchvision
jupyter notebook ResNet_CIFAR10.ipynb