From 781f512b01bc2324d7fdd11f0901f60571fc476f Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Mon, 3 Apr 2023 12:49:15 +0100 Subject: [PATCH] Fix mvitv2 docstring format (#7487) --- torchvision/models/video/mvit.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/torchvision/models/video/mvit.py b/torchvision/models/video/mvit.py index 6bde4a25016..973ff34cca2 100644 --- a/torchvision/models/video/mvit.py +++ b/torchvision/models/video/mvit.py @@ -765,11 +765,10 @@ def mvit_v1_b(*, weights: Optional[MViT_V1_B_Weights] = None, progress: bool = T @register_model() @handle_legacy_interface(weights=("pretrained", MViT_V2_S_Weights.KINETICS400_V1)) def mvit_v2_s(*, weights: Optional[MViT_V2_S_Weights] = None, progress: bool = True, **kwargs: Any) -> MViT: - """ - Constructs a small MViTV2 architecture from - `Multiscale Vision Transformers `__ - and `MViTv2: Improved Multiscale Vision Transformers for Classification and Detection -`__. + """Constructs a small MViTV2 architecture from + `Multiscale Vision Transformers `__ and + `MViTv2: Improved Multiscale Vision Transformers for Classification + and Detection `__. .. betastatus:: video module @@ -787,7 +786,7 @@ def mvit_v2_s(*, weights: Optional[MViT_V2_S_Weights] = None, progress: bool = T for more details about this class. .. autoclass:: torchvision.models.video.MViT_V2_S_Weights - :members: + :members: """ weights = MViT_V2_S_Weights.verify(weights)