Train CIFAR10 with PyTorch

Pytorch testbench PyTorch on the CIFAR10 dataset for lp batchnorm.

Prerequisites

• Python 3.6+
• PyTorch 1.0+

Implemented networks: VGG11, ResNet18, EfficientNetB0, GoogLeNet, DPN26, ShuffleNetG2, DenseNet121

To run the code, type: python main.py --net_name VGG --lr 0.01 --epochs 300 --batch_size 200 --exp FULL