Downloads · 30 days
0
mobilenet_v2_imagenette
alexsu52/mobilenet_v2_imagenette
mobilenet_v2_imagenette is a machine learning model from alexsu52. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
MobileNet V2 model from Torchvision fine-tuned for Imagenette dataset. Checkpoint trained for 100 epoches using https://github.com/alexsu52/mobilenetv2imagenette.
Access
Public
Updated Dec 15, 2022
Repo size
145 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin18.2 MB · 40%
At a glance
- License
- apache-2.0
- Access
- Public
- Created
- Dec 5, 2022
- Updated
- Dec 15, 2022
- SHA
- 64482e81
Model card
From the Hugging Face model README
MobileNet V2 model from Torchvision fine-tuned for Imagenette dataset. Checkpoint trained for 100 epoches using https://github.com/alexsu52/mobilenet_v2_imagenette.
Top-1 accuracy is 98.64%.
The main intent is to use it in samples and demos for model optimization. Here is the advantages:
- Imagenette can automatically downloaded and is quite small.
- MobileNet v2 is easy to train and lightweight model which is also representative and used in many public benchmarks.
Here is the code to load the checkpoint in PyTorch:
import torch
import torchvision.models as models
CHECKPOINT_URL = 'https://huggingface.co/alexsu52/mobilenet_v2_imagenette/resolve/main/pytorch_model.bin'
IMAGENETTE_CLASSES = 10
model = models.mobilenet_v2(num_classes=IMAGENETTE_CLASSES)
checkpoint = torch.hub.load_state_dict_from_url(CHECKPOINT_URL, progress=False)
model.load_state_dict(checkpoint['state_dict'])
- License
- apache-2.0
- Created
- Dec 5, 2022
- Updated
- Dec 15, 2022