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