Skip to content

Commit 3971255

Browse files
authored
Add missing paper link for mvitv2 (#7484)
1 parent 78c2719 commit 3971255

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

torchvision/models/video/mvit.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -767,7 +767,9 @@ def mvit_v1_b(*, weights: Optional[MViT_V1_B_Weights] = None, progress: bool = T
767767
def mvit_v2_s(*, weights: Optional[MViT_V2_S_Weights] = None, progress: bool = True, **kwargs: Any) -> MViT:
768768
"""
769769
Constructs a small MViTV2 architecture from
770-
`Multiscale Vision Transformers <https://arxiv.org/abs/2104.11227>`__.
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>`__.
771773
772774
.. betastatus:: video module
773775

0 commit comments

Comments
 (0)