Home

Awesome

Train CIFAR10 with PyTorch

I'm playing with PyTorch on the CIFAR10 dataset.

Prerequisites

Training

# Start training with: 
python main.py

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

Accuracy

ModelAcc.
VGG1692.64%
ResNet1893.02%
ResNet5093.62%
ResNet10193.75%
RegNetX_200MF94.24%
RegNetY_400MF94.29%
MobileNetV294.43%
ResNeXt29(32x4d)94.73%
ResNeXt29(2x64d)94.82%
SimpleDLA94.89%
DenseNet12195.04%
PreActResNet1895.11%
DPN9295.16%
DLA95.47%