[fbsync] Add revamped docs for Mobilenetv3 (#5908)

Summary:
* Start adding

* Finish adding docs

* Fix other bugs

* Update torchvision/models/mobilenetv3.py

* Update torchvision/models/mobilenetv3.py

Reviewed By: jdsgomes, NicolasHug

Differential Revision: D36095706

fbshipit-source-id: 4cacadb85cf008687246622e6c0d6a9380ef9ebe

Co-authored-by: Nicolas Hug <[email protected]>

Author: 2 people

docs/source/models/convnext.rst

@@ -1,5 +1,5 @@
 ConvNeXt
-======
+========
 
 .. currentmodule:: torchvision.models
 

docs/source/models/mobilenetv3.rst

@@ -0,0 +1,24 @@
+MobileNet V3
+============
+
+.. currentmodule:: torchvision.models
+
+The MobileNet V3 model is based on the `Searching for MobileNetV3 <https://arxiv.org/abs/1905.02244>`__ paper.
+
+
+Model builders
+--------------
+
+The following model builders can be used to instantiate a MobileNetV3 model, with or
+without pre-trained weights. All the model builders internally rely on the
+``torchvision.models.mobilenetv3.MobileNetV3`` base class. Please refer to the `source
+code
+<https://github.com/pytorch/vision/blob/main/torchvision/models/mobilenetv3.py>`_ for
+more details about this class.
+
+.. autosummary::
+    :toctree: generated/
+    :template: function.rst
+
+    mobilenet_v3_large
+    mobilenet_v3_small

docs/source/models_new.rst

@@ -43,6 +43,7 @@ weights:
    models/efficientnetv2
    models/googlenet
    models/mobilenetv2
+   models/mobilenetv3
    models/regnet
    models/resnet
    models/resnext

torchvision/models/mobilenetv3.py

@@ -362,11 +362,23 @@ def mobilenet_v3_large(
 ) -> MobileNetV3:
     """
     Constructs a large MobileNetV3 architecture from
-    `"Searching for MobileNetV3" <https://arxiv.org/abs/1905.02244>`_.
+    `Searching for MobileNetV3 <https://arxiv.org/abs/1905.02244>`__.
 
     Args:
-        weights (MobileNet_V3_Large_Weights, optional): The pretrained weights for the model
-        progress (bool): If True, displays a progress bar of the download to stderr
+        weights (:class:`~torchvision.models.MobileNet_V3_Large_Weights`, optional): The
+            pretrained weights to use. See
+            :class:`~torchvision.models.MobileNet_V3_Large_Weights` below for
+            more details, and possible values. By default, no pre-trained
+            weights are used.
+        progress (bool, optional): If True, displays a progress bar of the
+            download to stderr. Default is True.
+        **kwargs: parameters passed to the ``torchvision.models.resnet.MobileNetV3``
+            base class. Please refer to the `source code
+            <https://github.com/pytorch/vision/blob/main/torchvision/models/mobilenetv3.py>`_
+            for more details about this class.
+
+    .. autoclass:: torchvision.models.MobileNet_V3_Large_Weights
+        :members:
     """
     weights = MobileNet_V3_Large_Weights.verify(weights)
 
@@ -380,11 +392,23 @@ def mobilenet_v3_small(
 ) -> MobileNetV3:
     """
     Constructs a small MobileNetV3 architecture from
-    `"Searching for MobileNetV3" <https://arxiv.org/abs/1905.02244>`_.
+    `Searching for MobileNetV3 <https://arxiv.org/abs/1905.02244>`__.
 
     Args:
-        weights (MobileNet_V3_Small_Weights, optional): The pretrained weights for the model
-        progress (bool): If True, displays a progress bar of the download to stderr
+        weights (:class:`~torchvision.models.MobileNet_V3_Small_Weights`, optional): The
+            pretrained weights to use. See
+            :class:`~torchvision.models.MobileNet_V3_Small_Weights` below for
+            more details, and possible values. By default, no pre-trained
+            weights are used.
+        progress (bool, optional): If True, displays a progress bar of the
+            download to stderr. Default is True.
+        **kwargs: parameters passed to the ``torchvision.models.resnet.MobileNetV3``
+            base class. Please refer to the `source code
+            <https://github.com/pytorch/vision/blob/main/torchvision/models/mobilenetv3.py>`_
+            for more details about this class.
+
+    .. autoclass:: torchvision.models.MobileNet_V3_Small_Weights
+        :members:
     """
     weights = MobileNet_V3_Small_Weights.verify(weights)