Fix ViT and Resnext docs (pytorch#6150) (pytorch#6151)

datumbox · Jun 10, 2022 · 685b689 · 685b689
1 parent 072aa71
commit 685b689
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 15 deletions.
diff --git a/docs/source/models/resnext.rst b/docs/source/models/resnext.rst
@@ -23,3 +23,4 @@ more details about this class.
 
     resnext50_32x4d
     resnext101_32x8d
+    resnext101_64x4d
diff --git a/torchvision/models/vision_transformer.py b/torchvision/models/vision_transformer.py
@@ -603,16 +603,16 @@ def vit_b_16(*, weights: Optional[ViT_B_16_Weights] = None, progress: bool = Tru
     `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://arxiv.org/abs/2010.11929>`_.
 
     Args:
-        weights (:class:`~torchvision.models.vision_transformer.ViT_B_16_Weights`, optional): The pretrained
-            weights to use. See :class:`~torchvision.models.vision_transformer.ViT_B_16_Weights`
+        weights (:class:`~torchvision.models.ViT_B_16_Weights`, optional): The pretrained
+            weights to use. See :class:`~torchvision.models.ViT_B_16_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.vision_transformer.VisionTransformer``
             base class. Please refer to the `source code
             <https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_
             for more details about this class.
 
-    .. autoclass:: torchvision.models.vision_transformer.ViT_B_16_Weights
+    .. autoclass:: torchvision.models.ViT_B_16_Weights
         :members:
     """
     weights = ViT_B_16_Weights.verify(weights)
@@ -636,16 +636,16 @@ def vit_b_32(*, weights: Optional[ViT_B_32_Weights] = None, progress: bool = Tru
     `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://arxiv.org/abs/2010.11929>`_.
 
     Args:
-        weights (:class:`~torchvision.models.vision_transformer.ViT_B_32_Weights`, optional): The pretrained
-            weights to use. See :class:`~torchvision.models.vision_transformer.ViT_B_32_Weights`
+        weights (:class:`~torchvision.models.ViT_B_32_Weights`, optional): The pretrained
+            weights to use. See :class:`~torchvision.models.ViT_B_32_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.vision_transformer.VisionTransformer``
             base class. Please refer to the `source code
             <https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_
             for more details about this class.
 
-    .. autoclass:: torchvision.models.vision_transformer.ViT_B_32_Weights
+    .. autoclass:: torchvision.models.ViT_B_32_Weights
         :members:
     """
     weights = ViT_B_32_Weights.verify(weights)
@@ -669,16 +669,16 @@ def vit_l_16(*, weights: Optional[ViT_L_16_Weights] = None, progress: bool = Tru
     `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://arxiv.org/abs/2010.11929>`_.
 
     Args:
-        weights (:class:`~torchvision.models.vision_transformer.ViT_L_16_Weights`, optional): The pretrained
-            weights to use. See :class:`~torchvision.models.vision_transformer.ViT_L_16_Weights`
+        weights (:class:`~torchvision.models.ViT_L_16_Weights`, optional): The pretrained
+            weights to use. See :class:`~torchvision.models.ViT_L_16_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.vision_transformer.VisionTransformer``
             base class. Please refer to the `source code
             <https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_
             for more details about this class.
 
-    .. autoclass:: torchvision.models.vision_transformer.ViT_L_16_Weights
+    .. autoclass:: torchvision.models.ViT_L_16_Weights
         :members:
     """
     weights = ViT_L_16_Weights.verify(weights)
@@ -702,16 +702,16 @@ def vit_l_32(*, weights: Optional[ViT_L_32_Weights] = None, progress: bool = Tru
     `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://arxiv.org/abs/2010.11929>`_.
 
     Args:
-        weights (:class:`~torchvision.models.vision_transformer.ViT_L_32_Weights`, optional): The pretrained
-            weights to use. See :class:`~torchvision.models.vision_transformer.ViT_L_32_Weights`
+        weights (:class:`~torchvision.models.ViT_L_32_Weights`, optional): The pretrained
+            weights to use. See :class:`~torchvision.models.ViT_L_32_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.vision_transformer.VisionTransformer``
             base class. Please refer to the `source code
             <https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_
             for more details about this class.
 
-    .. autoclass:: torchvision.models.vision_transformer.ViT_L_32_Weights
+    .. autoclass:: torchvision.models.ViT_L_32_Weights
         :members:
     """
     weights = ViT_L_32_Weights.verify(weights)
@@ -734,16 +734,16 @@ def vit_h_14(*, weights: Optional[ViT_H_14_Weights] = None, progress: bool = Tru
     `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://arxiv.org/abs/2010.11929>`_.
 
     Args:
-        weights (:class:`~torchvision.models.vision_transformer.ViT_H_14_Weights`, optional): The pretrained
-            weights to use. See :class:`~torchvision.models.vision_transformer.ViT_H_14_Weights`
+        weights (:class:`~torchvision.models.ViT_H_14_Weights`, optional): The pretrained
+            weights to use. See :class:`~torchvision.models.ViT_H_14_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.vision_transformer.VisionTransformer``
             base class. Please refer to the `source code
             <https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_
             for more details about this class.
 
-    .. autoclass:: torchvision.models.vision_transformer.ViT_H_14_Weights
+    .. autoclass:: torchvision.models.ViT_H_14_Weights
         :members:
     """
     weights = ViT_H_14_Weights.verify(weights)