Skip to content

Commit

Permalink
Fix mvitv2 docstring format (#7487)
Browse files Browse the repository at this point in the history
  • Loading branch information
NicolasHug authored Apr 3, 2023
1 parent 3971255 commit 781f512
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions torchvision/models/video/mvit.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 <https://arxiv.org/abs/2104.11227>`__
and `MViTv2: Improved Multiscale Vision Transformers for Classification and Detection
<https://arxiv.org/abs/2112.01526>`__.
"""Constructs a small MViTV2 architecture from
`Multiscale Vision Transformers <https://arxiv.org/abs/2104.11227>`__ and
`MViTv2: Improved Multiscale Vision Transformers for Classification
and Detection <https://arxiv.org/abs/2112.01526>`__.
.. betastatus:: video module
Expand All @@ -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)

Expand Down

0 comments on commit 781f512

Please sign in to comment.