You can deploy this model and start earning money today!

alexsu52/
mobilenet_v2_imagenette

Name: mobilenet_v2_imagenette
Availability: InStock

pytorch

onnx

apache-2.0

Views

No views yet

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:

python

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'])