kuangliu/pytorch-cifar

95.47% on CIFAR10 with PyTorch

Python

6,444

27 commits

updated Feb 24, 2023

See the code

README

Train CIFAR10 with PyTorch

I'm playing with PyTorch on the CIFAR10 dataset.

Prerequisites

Python 3.6+
PyTorch 1.0+

Training

# Start training with: 
python main.py

# You can manually resume the training with: 
python main.py --resume --lr=0.01

Accuracy

Model	Acc.
VGG16	92.64%
ResNet18	93.02%
ResNet50	93.62%
ResNet101	93.75%
RegNetX_200MF	94.24%
RegNetY_400MF	94.29%
MobileNetV2	94.43%
ResNeXt29(32x4d)	94.73%
ResNeXt29(2x64d)	94.82%
SimpleDLA	94.89%
DenseNet121	95.04%
PreActResNet18	95.11%
DPN92	95.16%
DLA	95.47%

pytorch

Project health

Overall

2.0/10

Maintained

0/10

Code review

0/10

OpenSSF Scorecard

Significant stargazers

Urchade Zaratiana

232 followers · starred Jun 2021

Wilson Yan

222 followers · starred Jun 2021

Hao Shao

148 followers · starred Jan 2019

Hugo Laurençon

59 followers · starred Feb 2021

Similar repos

akamaster/pytorch_resnet_cifar10

Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the…

1,371

BIGBALLON/cifar-10-cnn

Play deep learning with CIFAR datasets

840

weiaicunzai/awesome-image-classification

A curated list of deep learning image classification papers and codes

3,067

Cadene/pretrained-models.pytorch

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception,…

9,101

rasbt/deeplearning-models

A collection of various deep learning architectures, models, and tips

17,612

gmalivenko/awesome-computer-vision-models

A list of popular deep learning models related to classification, segmentation and detection…

545

facebookresearch/ConvNeXt

Code release for ConvNeXt model

6,408

KaimingHe/deep-residual-networks

Deep Residual Learning for Image Recognition

6,762

kuangliu/pytorch-cifar

95.47% on CIFAR10 with PyTorch

Python

6,444

27 commits

updated Feb 24, 2023

See the code

README

Train CIFAR10 with PyTorch

I'm playing with PyTorch on the CIFAR10 dataset.

Prerequisites

Python 3.6+
PyTorch 1.0+

Training

# Start training with: 
python main.py

# You can manually resume the training with: 
python main.py --resume --lr=0.01

Accuracy

Model	Acc.
VGG16	92.64%
ResNet18	93.02%
ResNet50	93.62%
ResNet101	93.75%
RegNetX_200MF	94.24%
RegNetY_400MF	94.29%
MobileNetV2	94.43%
ResNeXt29(32x4d)	94.73%
ResNeXt29(2x64d)	94.82%
SimpleDLA	94.89%
DenseNet121	95.04%
PreActResNet18	95.11%
DPN92	95.16%
DLA	95.47%

pytorch

Project health

Overall

2.0/10

Maintained

0/10

Code review

0/10

OpenSSF Scorecard

Significant stargazers

Urchade Zaratiana

232 followers · starred Jun 2021

Wilson Yan

222 followers · starred Jun 2021

Hao Shao

148 followers · starred Jan 2019

Hugo Laurençon

59 followers · starred Feb 2021

Similar repos

akamaster/pytorch_resnet_cifar10

Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the…

1,371

BIGBALLON/cifar-10-cnn

Play deep learning with CIFAR datasets

840

weiaicunzai/awesome-image-classification

A curated list of deep learning image classification papers and codes

3,067

Cadene/pretrained-models.pytorch

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception,…

9,101

rasbt/deeplearning-models

A collection of various deep learning architectures, models, and tips

17,612

gmalivenko/awesome-computer-vision-models

A list of popular deep learning models related to classification, segmentation and detection…

545

facebookresearch/ConvNeXt

Code release for ConvNeXt model

6,408

KaimingHe/deep-residual-networks

Deep Residual Learning for Image Recognition

6,762