duanjiaqi/pytorch-cifar10-test

test cluster

Train CIFAR10 with PyTorch

I'm playing with PyTorch on the CIFAR10 dataset.

Prerequisites

• Python 3.6+
• PyTorch 1.0+

Accuracy

Model	Acc.
ResNet18	93.02%
ResNet50	93.62%
ResNet101	93.75%

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