Skip to content

Latest commit

 

History

History
69 lines (28 loc) · 1.15 KB

README.md

File metadata and controls

69 lines (28 loc) · 1.15 KB

CIFAR10-100-fast-training

This implementation is for cifar10/100 fast training base on (https://github.com/davidcpage/cifar10-fast)

Please check demo.ipynb.

You can run 70 epochs rather than 350 (https://github.com/kuangliu/pytorch-cifar) or 200 (https://github.com/weiaicunzai/pytorch-cifar100) epochs.

Requirements:

    pytorch
    apex (https://nvidia.github.io/apex/)
    numpy

Results:

Those models from https://github.com/weiaicunzai/pytorch-cifar100/tree/master/models and https://github.com/kuangliu/pytorch-cifar/tree/master/models

Model CIFAR-10 accuracy CIFAR-100 accuracy
ResNet18 95.32 76.75
ResNet34 95.57 77.48
ResNet50 95.62 77.66
MobileNet 92.25 60.35
MobileNetV2 93.69 62.56
DenseNet-cifar 94.7 72.61
DenseNet121 95.1 76.97
DenseNet201 94.89 77.28
Wide-ResNet40 95.13 74.79
Wide-ResNet16 95.4 78.52
Wide-ResNet28 96.21 79.86
VGG11 92.41 70.07
VGG16 94.27 72.68
VGG19 94.22 71.11
GoogleNet 95.35 79.24
InceptionV3 95.55 79.29