Skip to content

Latest commit

 

History

History
15 lines (12 loc) · 418 Bytes

README.md

File metadata and controls

15 lines (12 loc) · 418 Bytes

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