From 70ed33fe7f3bba1341570497584500ef393f73d1 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Fri, 31 Mar 2023 15:07:19 +0100 Subject: [PATCH] Add missing paper link for mvitv2 --- torchvision/models/video/mvit.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/torchvision/models/video/mvit.py b/torchvision/models/video/mvit.py index d3be0374012..6bde4a25016 100644 --- a/torchvision/models/video/mvit.py +++ b/torchvision/models/video/mvit.py @@ -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: """ Constructs a small MViTV2 architecture from - `Multiscale Vision Transformers `__. + `Multiscale Vision Transformers `__ + and `MViTv2: Improved Multiscale Vision Transformers for Classification and Detection +`__. .. betastatus:: video module