Skip to content

WeihaoZhuang/cifar10-100-fast-training

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published