Skip to content

Commit 781f512

Browse files
authored
Fix mvitv2 docstring format (#7487)
1 parent 3971255 commit 781f512

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

torchvision/models/video/mvit.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -765,11 +765,10 @@ def mvit_v1_b(*, weights: Optional[MViT_V1_B_Weights] = None, progress: bool = T
765765
@register_model()
766766
@handle_legacy_interface(weights=("pretrained", MViT_V2_S_Weights.KINETICS400_V1))
767767
def mvit_v2_s(*, weights: Optional[MViT_V2_S_Weights] = None, progress: bool = True, **kwargs: Any) -> MViT:
768-
"""
769-
Constructs a small MViTV2 architecture from
770-
`Multiscale Vision Transformers <https://arxiv.org/abs/2104.11227>`__
771-
and `MViTv2: Improved Multiscale Vision Transformers for Classification and Detection
772-
<https://arxiv.org/abs/2112.01526>`__.
768+
"""Constructs a small MViTV2 architecture from
769+
`Multiscale Vision Transformers <https://arxiv.org/abs/2104.11227>`__ and
770+
`MViTv2: Improved Multiscale Vision Transformers for Classification
771+
and Detection <https://arxiv.org/abs/2112.01526>`__.
773772
774773
.. betastatus:: video module
775774
@@ -787,7 +786,7 @@ def mvit_v2_s(*, weights: Optional[MViT_V2_S_Weights] = None, progress: bool = T
787786
for more details about this class.
788787
789788
.. autoclass:: torchvision.models.video.MViT_V2_S_Weights
790-
:members:
789+
:members:
791790
"""
792791
weights = MViT_V2_S_Weights.verify(weights)
793792

0 commit comments

Comments
 (0)