Skip to content

Training of MobileNet v2 from Torchvision on Food-101 dataset

License

Notifications You must be signed in to change notification settings

AlexKoff88/mobilenetv2_food101

Repository files navigation

MobileNet v2 PyTroch on Food101

Training of MobileNet v2 from Torchvision on Food101 dataset

Run training:

python train.py --epochs 30 -j 6 -b 256 --lr 0.001 --pretrained --wd 10e-5 dataset

Pretrained model:

You can download pretrained model from here. Top-1 accuracy is 76.3%.

Export to ONNX:

python export.py model_best.pth.tar

About

Training of MobileNet v2 from Torchvision on Food-101 dataset

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages