From acd63b39f4c2b3094c520228adce22b861325bca Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Mon, 16 May 2022 13:44:40 +0100 Subject: [PATCH] Fix ConvNext weight links --- torchvision/models/convnext.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/torchvision/models/convnext.py b/torchvision/models/convnext.py index 64f832a2a58..f2c7ee1e03a 100644 --- a/torchvision/models/convnext.py +++ b/torchvision/models/convnext.py @@ -289,7 +289,7 @@ def convnext_tiny(*, weights: Optional[ConvNeXt_Tiny_Weights] = None, progress: `_ for more details about this class. - .. autoclass:: torchvision.models.convnext.ConvNeXt_Tiny_Weights + .. autoclass:: torchvision.models.ConvNeXt_Tiny_Weights :members: """ weights = ConvNeXt_Tiny_Weights.verify(weights) @@ -321,7 +321,7 @@ def convnext_small( `_ for more details about this class. - .. autoclass:: torchvision.models.convnext.ConvNeXt_Small_Weights + .. autoclass:: torchvision.models.ConvNeXt_Small_Weights :members: """ weights = ConvNeXt_Small_Weights.verify(weights) @@ -351,7 +351,7 @@ def convnext_base(*, weights: Optional[ConvNeXt_Base_Weights] = None, progress: `_ for more details about this class. - .. autoclass:: torchvision.models.convnext.ConvNeXt_Base_Weights + .. autoclass:: torchvision.models.ConvNeXt_Base_Weights :members: """ weights = ConvNeXt_Base_Weights.verify(weights) @@ -383,7 +383,7 @@ def convnext_large( `_ for more details about this class. - .. autoclass:: torchvision.models.convnext.ConvNeXt_Large_Weights + .. autoclass:: torchvision.models.ConvNeXt_Large_Weights :members: """ weights = ConvNeXt_Large_Weights.verify(weights)