Skip to content

GitLab

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

Fix ConvNext weight links (#6023)

parent a161098e
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
torchvision/models/convnext.py
View file @ 8e5844fc
...@@ -289,7 +289,7 @@ def convnext_tiny(*, weights: Optional[ConvNeXt_Tiny_Weights] = None, progress: ...@@ -289,7 +289,7 @@ def convnext_tiny(*, weights: Optional[ConvNeXt_Tiny_Weights] = None, progress:
<https://github.com/pytorch/vision/blob/main/torchvision/models/convnext.py>`_ <https://github.com/pytorch/vision/blob/main/torchvision/models/convnext.py>`_
for more details about this class. for more details about this class.
.. autoclass:: torchvision.models.convnext.ConvNeXt_Tiny_Weights .. autoclass:: torchvision.models.ConvNeXt_Tiny_Weights
:members: :members:
""" """
weights = ConvNeXt_Tiny_Weights.verify(weights) weights = ConvNeXt_Tiny_Weights.verify(weights)
...@@ -321,7 +321,7 @@ def convnext_small( ...@@ -321,7 +321,7 @@ def convnext_small(
<https://github.com/pytorch/vision/blob/main/torchvision/models/convnext.py>`_ <https://github.com/pytorch/vision/blob/main/torchvision/models/convnext.py>`_
for more details about this class. for more details about this class.
.. autoclass:: torchvision.models.convnext.ConvNeXt_Small_Weights .. autoclass:: torchvision.models.ConvNeXt_Small_Weights
:members: :members:
""" """
weights = ConvNeXt_Small_Weights.verify(weights) weights = ConvNeXt_Small_Weights.verify(weights)
...@@ -351,7 +351,7 @@ def convnext_base(*, weights: Optional[ConvNeXt_Base_Weights] = None, progress: ...@@ -351,7 +351,7 @@ def convnext_base(*, weights: Optional[ConvNeXt_Base_Weights] = None, progress:
<https://github.com/pytorch/vision/blob/main/torchvision/models/convnext.py>`_ <https://github.com/pytorch/vision/blob/main/torchvision/models/convnext.py>`_
for more details about this class. for more details about this class.
.. autoclass:: torchvision.models.convnext.ConvNeXt_Base_Weights .. autoclass:: torchvision.models.ConvNeXt_Base_Weights
:members: :members:
""" """
weights = ConvNeXt_Base_Weights.verify(weights) weights = ConvNeXt_Base_Weights.verify(weights)
...@@ -383,7 +383,7 @@ def convnext_large( ...@@ -383,7 +383,7 @@ def convnext_large(
<https://github.com/pytorch/vision/blob/main/torchvision/models/convnext.py>`_ <https://github.com/pytorch/vision/blob/main/torchvision/models/convnext.py>`_
for more details about this class. for more details about this class.
.. autoclass:: torchvision.models.convnext.ConvNeXt_Large_Weights .. autoclass:: torchvision.models.ConvNeXt_Large_Weights
:members: :members:
""" """
weights = ConvNeXt_Large_Weights.verify(weights) weights = ConvNeXt_Large_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