Skip to content

MarioJayakumar/pytorch-cifar

This branch is 86 commits ahead of, 9 commits behind kuangliu/pytorch-cifar:master.

Folders and files

NameName
Last commit message
Last commit date
Feb 10, 2021
Jun 12, 2020
Aug 29, 2020
Nov 16, 2017
Apr 15, 2020
Aug 26, 2020
Jan 31, 2021
Jan 31, 2021
Dec 30, 2020
Dec 30, 2020
Jun 18, 2020
Jun 18, 2020
Jan 16, 2021
Jan 16, 2021
Jan 23, 2021
Jan 23, 2021
Jan 23, 2021
Jan 23, 2021
Feb 2, 2021
Aug 26, 2020
Aug 26, 2020
Aug 29, 2020
Aug 29, 2020
Aug 29, 2020
Aug 29, 2020
Aug 29, 2020
Sep 1, 2020
Sep 1, 2020
Sep 1, 2020
Jan 31, 2021
Jan 27, 2021
Aug 24, 2020
Jan 31, 2021
Jan 31, 2021
Jan 31, 2021
Jan 31, 2021
Jan 31, 2021
Jan 31, 2021
Jan 31, 2021
Jan 31, 2021
Jan 31, 2021
Jan 31, 2021
Jun 15, 2020
Jun 15, 2020
Jan 27, 2021
Jan 27, 2021
Jan 27, 2021
Jan 27, 2021
Jan 27, 2021
Jan 27, 2021
Jan 27, 2021
Jan 28, 2021
Jan 27, 2021
Jan 27, 2021
Dec 27, 2020

Repository files navigation

Train CIFAR10 with PyTorch

I'm playing with PyTorch on the CIFAR10 dataset.

Prerequisites

  • Python 3.6+
  • PyTorch 1.0+

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%
DenseNet121 95.04%
PreActResNet18 95.11%
DPN92 95.16%

Learning rate adjustment

I manually change the lr during training:

  • 0.1 for epoch [0,150)
  • 0.01 for epoch [150,250)
  • 0.001 for epoch [250,350)

Resume the training with python main.py --resume --lr=0.01

About

95.16% on CIFAR10 with PyTorch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 86.1%
  • Shell 13.9%