Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
781f512b
"git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "c90f2be0be205ca8fd8cf0981dd9226d90a6949d"
Unverified
Commit
781f512b
authored
Apr 03, 2023
by
Nicolas Hug
Committed by
GitHub
Apr 03, 2023
Browse files
Fix mvitv2 docstring format (#7487)
parent
39712558
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
6 deletions
+5
-6
torchvision/models/video/mvit.py
torchvision/models/video/mvit.py
+5
-6
No files found.
torchvision/models/video/mvit.py
View file @
781f512b
...
@@ -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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment