pytorch · yassineAlouini · Aug 18, 2022 · Aug 25, 2022 · Aug 31, 2022 · Nov 2, 2022
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -18,7 +18,7 @@ repos:
           - black == 22.3.0
           - usort == 1.0.2
 
-  - repo: https://gitlab.com/pycqa/flake8
+  - repo: https://github.com/pycqa/flake8
     rev: 3.9.2
     hooks:
       - id: flake8

diff --git a/docs/source/models/mobilevit.rst b/docs/source/models/mobilevit.rst
@@ -0,0 +1,27 @@
+..
+ _TODO: Update the documentation with the correct links...
+
+
+MobileViT
+===========
+
+.. currentmodule:: torchvision.models
+
+The MobileViT model is based on the `"MobileViT: Light-Weight, General-Purpose, and Mobile-Friendly Vision Transfomer" <https://arxiv.org/abs/2110.02178>`__ paper.
+
+
+Model builders
+--------------
+
+The following model builders can be used to instantiate a MobileViT model, with or
+without pre-trained weights. All the model builders internally rely on the
+``torchvision.models.mobilevit.MobileViT`` base class. Please refer to the `source
+code
+<https://github.com/pytorch/vision/blob/main/torchvision/models/mobilevit.py>`_ for
+more details about this class.
+
+.. autosummary::
+    :toctree: generated/
+    :template: function.rst
+
+    mobilevit
diff --git a/run.sh b/run.sh
@@ -0,0 +1 @@
+python setup.py develop && torchrun --nproc_per_node=8 vision-1/references/classification/train.py --model mobile_vit_xxs
diff --git a/torchvision/models/__init__.py b/torchvision/models/__init__.py
@@ -13,5 +13,6 @@
 from .vgg import *
 from .vision_transformer import *
 from .swin_transformer import *
+from .mobilevit import *
 from . import detection, optical_flow, quantization, segmentation, video
 from ._api import get_model, get_model_weights, get_weight, list_models
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		python setup.py develop && torchrun --nproc_per_node=8 vision-1/references/classification/train.py --model mobile_vit_xxs