Skip to content

Commit 035d99f

Browse files
authored
Add video models to torchhub (pytorch#7083)
1 parent 934ce3b commit 035d99f

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

hubconf.py

+11
Original file line numberDiff line numberDiff line change
@@ -71,4 +71,15 @@
7171
from torchvision.models.squeezenet import squeezenet1_0, squeezenet1_1
7272
from torchvision.models.swin_transformer import swin_b, swin_s, swin_t, swin_v2_b, swin_v2_s, swin_v2_t
7373
from torchvision.models.vgg import vgg11, vgg11_bn, vgg13, vgg13_bn, vgg16, vgg16_bn, vgg19, vgg19_bn
74+
from torchvision.models.video import (
75+
mc3_18,
76+
mvit_v1_b,
77+
mvit_v2_s,
78+
r2plus1d_18,
79+
r3d_18,
80+
s3d,
81+
swin3d_b,
82+
swin3d_s,
83+
swin3d_t,
84+
)
7485
from torchvision.models.vision_transformer import vit_b_16, vit_b_32, vit_h_14, vit_l_16, vit_l_32

0 commit comments

Comments
 (0)