cnn-cifar10-classifier v0.1.0
Small PyTorch CNN baseline for CIFAR-10 image classification.
This release contains the trained v0.1.0 checkpoint from the GitHub project:
Results
| Split | Metric | Value |
|---|
| test | accuracy | 0.7857 |
The checkpoint was selected by best validation/test accuracy during a 20 epoch run.
Training summary:
- Dataset:
uoft-cs/cifar10
- Config:
plain_text
- Epochs: 20
- Batch size: 256
- Optimizer: SGD, momentum 0.9, weight decay 0.0005
- Learning rate: 0.01
- Augmentation: random crop with padding 4, random horizontal flip
- AMP: enabled
- GPU used: NVIDIA GeForce RTX 5070 Ti
- PyTorch: 2.11.0+cu128
Files
1checkpoints/best_model.pth
2checkpoints/last_model.pth
3outputs/training_metrics.json
4outputs/training_curves.png
5outputs/confusion_matrix.png
6outputs/demo_predictions.png
7logs/train_20260602_044856.log
8logs/evaluate_20260602_045254.log
9logs/demo_20260602_045311.log
10runs/cifar10_cnn_20260602_044901/events.out.tfevents...
11release_summary.json
12manifest.sha256
Usage
Clone the project code first:
1git clone git@github.com:diverHansun/cnn-cifar10-classifier.git
2cd cnn-cifar10-classifier
Install dependencies with a CUDA-compatible PyTorch build for your machine, then download this checkpoint:
hf download diverWayne/cnn-cifar10-classifier checkpoints/best_model.pth --local-dir .Evaluate:
python evaluate.py --checkpoint checkpoints/best_model.pth --device cudaRun the demo grid:
python demo.py --checkpoint checkpoints/best_model.pth --samples 16 --device cudaPredict one image:
python predict.py --image demo_images/your_image.png --checkpoint checkpoints/best_model.pth --device cudaLimitations
This is a simple hand-written CNN baseline trained on CIFAR-10 32x32 images. It supports the 10 CIFAR-10 classes only:
airplane, automobile, bird, cat, deer, dog, frog, horse, ship, truckIt can classify arbitrary images after resizing to 32x32, but reliability outside CIFAR-10-like images is limited.
The strongest observed confusions are between visually similar categories such as cat and dog, bird and deer/dog, and airplane and ship.
Dataset
The training data is not redistributed in this model repository. It is loaded from the public Hugging Face dataset uoft-cs/cifar10.
