Zero to Mastery Learn PyTorch for Deep Learning exercise - Food 101 Classifier in PyTorch, algorithm implemented as follows:
- TinyVGG
- EffNetB2 Transfer Learning
- ViT-B/16 Transfer Learning
Dataset: Food101
- use 20% of Food101 data
# --m: model type(default/tinyvgg: TinyVGG, vit: ViT-B/16, effnet:EffNetB2)
# --e: epoch (default:2)
python main.py --m "vit" --e 3
Use gradio to demo model(EffNetB2) result, please click link to give it a try!