VGG, ResNet, MobileNet, ShuffleNet, GhostNet... for CIFAR10 or CIFAR100 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 |
-
Notifications
You must be signed in to change notification settings - Fork 1
tingggggg/pytorch-classification
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
VGG, ResNet, MobileNet, ShuffleNet, GhostNet... for CIFAR10 or CIFAR100 Training
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published