Skip to content

GitLab

Unverified Commit 781f512b authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix mvitv2 docstring format (#7487)

parent 39712558
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 6 deletions
torchvision/models/video/mvit.py
View file @ 781f512b
...@@ -765,11 +765,10 @@ def mvit_v1_b(*, weights: Optional[MViT_V1_B_Weights] = None, progress: bool = T ...@@ -765,11 +765,10 @@ def mvit_v1_b(*, weights: Optional[MViT_V1_B_Weights] = None, progress: bool = T
@register_model() @register_model()
@handle_legacy_interface(weights=("pretrained", MViT_V2_S_Weights.KINETICS400_V1)) @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: def mvit_v2_s(*, weights: Optional[MViT_V2_S_Weights] = None, progress: bool = True, **kwargs: Any) -> MViT:
""" """Constructs a small MViTV2 architecture from
Constructs a small MViTV2 architecture from `Multiscale Vision Transformers <https://arxiv.org/abs/2104.11227>`__ and
`Multiscale Vision Transformers <https://arxiv.org/abs/2104.11227>`__ `MViTv2: Improved Multiscale Vision Transformers for Classification
and `MViTv2: Improved Multiscale Vision Transformers for Classification and Detection and Detection <https://arxiv.org/abs/2112.01526>`__.
<https://arxiv.org/abs/2112.01526>`__.
.. betastatus:: video module .. betastatus:: video module
......
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