Pytorch-CIFAR10
Pytorch-CIFAR10 copied to clipboard
Published
20 hours ago •
kanedaaaa
kanedaaaa
Image classification models on CIFAR10 dataset using pytorch
pytorch-cifar10
Classifying CIFAR10 dataset with popular DL computer vision models.
Requirements
CUDA and proper NVIDIA drivers (optional, only if Nvidia GPU is available)
Instructions
python main.py
model folder contains net architectures, just uncomment the preferred one in main.py
Current stats
| Model | GPU | Accuracy | Training Time |
|---|---|---|---|
| LeNet5 | Tesla T4 | 67.15% | ... |
| ResNet | Tesla T4 | 76.14% | 21 min |
| VGG16 | Tesla T4 | 78.60% | 6 min |
| VGG19 | Tesla T4 | 78.51% | 7 min |
Extras
project is no longer maintained
kanedaaaa