readme: ---
  license: apache-2.0
  ---
  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'])
  ```
  