"git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "c90f2be0be205ca8fd8cf0981dd9226d90a6949d"
Unverified Commit 781f512b authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix mvitv2 docstring format (#7487)

parent 39712558
...@@ -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
...@@ -787,7 +786,7 @@ def mvit_v2_s(*, weights: Optional[MViT_V2_S_Weights] = None, progress: bool = T ...@@ -787,7 +786,7 @@ def mvit_v2_s(*, weights: Optional[MViT_V2_S_Weights] = None, progress: bool = T
for more details about this class. for more details about this class.
.. autoclass:: torchvision.models.video.MViT_V2_S_Weights .. autoclass:: torchvision.models.video.MViT_V2_S_Weights
:members: :members:
""" """
weights = MViT_V2_S_Weights.verify(weights) weights = MViT_V2_S_Weights.verify(weights)
......
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