Skip to content

GitLab

Unverified Commit fee6d12c authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix ViT and Resnext docs (#6150)

parent 5f6e22dc
Hide whitespace changes
Inline Side-by-side
Showing with 16 additions and 15 deletions
docs/source/models/resnext.rst
View file @ fee6d12c
...@@ -23,3 +23,4 @@ more details about this class. ...@@ -23,3 +23,4 @@ more details about this class.
resnext50_32x4d resnext50_32x4d
resnext101_32x8d resnext101_32x8d
resnext101_64x4d
torchvision/models/vision_transformer.py
View file @ fee6d12c
...@@ -603,8 +603,8 @@ def vit_b_16(*, weights: Optional[ViT_B_16_Weights] = None, progress: bool = Tru ...@@ -603,8 +603,8 @@ 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>`_. `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://arxiv.org/abs/2010.11929>`_.
Args: Args:
weights (:class:`~torchvision.models.vision_transformer.ViT_B_16_Weights`, optional): The pretrained weights (:class:`~torchvision.models.ViT_B_16_Weights`, optional): The pretrained
weights to use. See :class:`~torchvision.models.vision_transformer.ViT_B_16_Weights` 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. 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. 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`` **kwargs: parameters passed to the ``torchvision.models.vision_transformer.VisionTransformer``
...@@ -612,7 +612,7 @@ def vit_b_16(*, weights: Optional[ViT_B_16_Weights] = None, progress: bool = Tru ...@@ -612,7 +612,7 @@ def vit_b_16(*, weights: Optional[ViT_B_16_Weights] = None, progress: bool = Tru
<https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_ <https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_
for more details about this class. for more details about this class.
.. autoclass:: torchvision.models.vision_transformer.ViT_B_16_Weights .. autoclass:: torchvision.models.ViT_B_16_Weights
:members: :members:
""" """
weights = ViT_B_16_Weights.verify(weights) weights = ViT_B_16_Weights.verify(weights)
...@@ -636,8 +636,8 @@ def vit_b_32(*, weights: Optional[ViT_B_32_Weights] = None, progress: bool = Tru ...@@ -636,8 +636,8 @@ 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>`_. `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://arxiv.org/abs/2010.11929>`_.
Args: Args:
weights (:class:`~torchvision.models.vision_transformer.ViT_B_32_Weights`, optional): The pretrained weights (:class:`~torchvision.models.ViT_B_32_Weights`, optional): The pretrained
weights to use. See :class:`~torchvision.models.vision_transformer.ViT_B_32_Weights` 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. 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. 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`` **kwargs: parameters passed to the ``torchvision.models.vision_transformer.VisionTransformer``
...@@ -645,7 +645,7 @@ def vit_b_32(*, weights: Optional[ViT_B_32_Weights] = None, progress: bool = Tru ...@@ -645,7 +645,7 @@ def vit_b_32(*, weights: Optional[ViT_B_32_Weights] = None, progress: bool = Tru
<https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_ <https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_
for more details about this class. for more details about this class.
.. autoclass:: torchvision.models.vision_transformer.ViT_B_32_Weights .. autoclass:: torchvision.models.ViT_B_32_Weights
:members: :members:
""" """
weights = ViT_B_32_Weights.verify(weights) weights = ViT_B_32_Weights.verify(weights)
...@@ -669,8 +669,8 @@ def vit_l_16(*, weights: Optional[ViT_L_16_Weights] = None, progress: bool = Tru ...@@ -669,8 +669,8 @@ 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>`_. `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://arxiv.org/abs/2010.11929>`_.
Args: Args:
weights (:class:`~torchvision.models.vision_transformer.ViT_L_16_Weights`, optional): The pretrained weights (:class:`~torchvision.models.ViT_L_16_Weights`, optional): The pretrained
weights to use. See :class:`~torchvision.models.vision_transformer.ViT_L_16_Weights` 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. 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. 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`` **kwargs: parameters passed to the ``torchvision.models.vision_transformer.VisionTransformer``
...@@ -678,7 +678,7 @@ def vit_l_16(*, weights: Optional[ViT_L_16_Weights] = None, progress: bool = Tru ...@@ -678,7 +678,7 @@ def vit_l_16(*, weights: Optional[ViT_L_16_Weights] = None, progress: bool = Tru
<https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_ <https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_
for more details about this class. for more details about this class.
.. autoclass:: torchvision.models.vision_transformer.ViT_L_16_Weights .. autoclass:: torchvision.models.ViT_L_16_Weights
:members: :members:
""" """
weights = ViT_L_16_Weights.verify(weights) weights = ViT_L_16_Weights.verify(weights)
...@@ -702,8 +702,8 @@ def vit_l_32(*, weights: Optional[ViT_L_32_Weights] = None, progress: bool = Tru ...@@ -702,8 +702,8 @@ 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>`_. `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://arxiv.org/abs/2010.11929>`_.
Args: Args:
weights (:class:`~torchvision.models.vision_transformer.ViT_L_32_Weights`, optional): The pretrained weights (:class:`~torchvision.models.ViT_L_32_Weights`, optional): The pretrained
weights to use. See :class:`~torchvision.models.vision_transformer.ViT_L_32_Weights` 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. 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. 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`` **kwargs: parameters passed to the ``torchvision.models.vision_transformer.VisionTransformer``
...@@ -711,7 +711,7 @@ def vit_l_32(*, weights: Optional[ViT_L_32_Weights] = None, progress: bool = Tru ...@@ -711,7 +711,7 @@ def vit_l_32(*, weights: Optional[ViT_L_32_Weights] = None, progress: bool = Tru
<https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_ <https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_
for more details about this class. for more details about this class.
.. autoclass:: torchvision.models.vision_transformer.ViT_L_32_Weights .. autoclass:: torchvision.models.ViT_L_32_Weights
:members: :members:
""" """
weights = ViT_L_32_Weights.verify(weights) weights = ViT_L_32_Weights.verify(weights)
...@@ -734,8 +734,8 @@ def vit_h_14(*, weights: Optional[ViT_H_14_Weights] = None, progress: bool = Tru ...@@ -734,8 +734,8 @@ 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>`_. `An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale <https://arxiv.org/abs/2010.11929>`_.
Args: Args:
weights (:class:`~torchvision.models.vision_transformer.ViT_H_14_Weights`, optional): The pretrained weights (:class:`~torchvision.models.ViT_H_14_Weights`, optional): The pretrained
weights to use. See :class:`~torchvision.models.vision_transformer.ViT_H_14_Weights` 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. 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. 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`` **kwargs: parameters passed to the ``torchvision.models.vision_transformer.VisionTransformer``
...@@ -743,7 +743,7 @@ def vit_h_14(*, weights: Optional[ViT_H_14_Weights] = None, progress: bool = Tru ...@@ -743,7 +743,7 @@ def vit_h_14(*, weights: Optional[ViT_H_14_Weights] = None, progress: bool = Tru
<https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_ <https://github.com/pytorch/vision/blob/main/torchvision/models/vision_transformer.py>`_
for more details about this class. for more details about this class.
.. autoclass:: torchvision.models.vision_transformer.ViT_H_14_Weights .. autoclass:: torchvision.models.ViT_H_14_Weights
:members: :members:
""" """
weights = ViT_H_14_Weights.verify(weights) weights = ViT_H_14_Weights.verify(weights)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment