lapida commited on
Commit
21f95af
1 Parent(s): eddf9a4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -0
README.md CHANGED
@@ -1,3 +1,12 @@
1
  ---
2
  license: apache-2.0
 
 
3
  ---
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ datasets:
4
+ - imagenet-1k
5
  ---
6
+ # Model card for pytorch_mobilevit_xs
7
+
8
+ A PyTorch implementation of MobileViT-XS model, based on [Timm](https://github.com/huggingface/pytorch-image-models).
9
+
10
+ The model weights were taken from [timm/mobilevit_xs.cvnets_in1k](https://huggingface.co/timm/mobilevit_xs.cvnets_in1k).
11
+
12
+ For further insights into the model's implementation details, please refer to [MCT Model Garden - MobileViT-XS](https://github.com/sony/model_optimization/tree/main/tutorials/mct_model_garden/models_pytorch/mobilevit_xs)