Classifying CIFAR10 dataset with popular DL computer vision models.
CUDA and proper NVIDIA drivers (optional, only if Nvidia GPU is available)
python main.py
model folder contains net architectures, just uncomment the preferred one in main.py
| 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 |
project is no longer maintained