Skip to content

GitLab

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

Add missing paper link for mvitv2 (#7484)

parent 78c27197
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
torchvision/models/video/mvit.py
View file @ 39712558
...@@ -767,7 +767,9 @@ def mvit_v1_b(*, weights: Optional[MViT_V1_B_Weights] = None, progress: bool = T ...@@ -767,7 +767,9 @@ def mvit_v1_b(*, weights: Optional[MViT_V1_B_Weights] = None, progress: bool = T
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>`__. `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 .. 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