pytorch-classification

VGG, ResNet, MobileNet, ShuffleNet, GhostNet... for CIFAR10 or CIFAR100 Training

Training

python main.py --net vgg16 --num_classes 100

--net	--num_classes
vgg11,13,16 or 19	10
resnet18,34,50,101 or 152	100
resnext
mobilenet
shufflenetv1
efficientnet
ghostnet