Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "f3c89cc696cb77a3d453d93f06b637b202316af1"
Unverified Commit 035d99f1 authored by Aditya Oke's avatar Aditya Oke Committed by GitHub
Browse files

Add video models to torchhub (#7083)

parent 934ce3b8
Hide whitespace changes
Inline Side-by-side
Showing with 11 additions and 0 deletions
hubconf.py
View file @ 035d99f1
...@@ -71,4 +71,15 @@ from torchvision.models.shufflenetv2 import ( ...@@ -71,4 +71,15 @@ from torchvision.models.shufflenetv2 import (
from torchvision.models.squeezenet import squeezenet1_0, squeezenet1_1 from torchvision.models.squeezenet import squeezenet1_0, squeezenet1_1
from torchvision.models.swin_transformer import swin_b, swin_s, swin_t, swin_v2_b, swin_v2_s, swin_v2_t from torchvision.models.swin_transformer import swin_b, swin_s, swin_t, swin_v2_b, swin_v2_s, swin_v2_t
from torchvision.models.vgg import vgg11, vgg11_bn, vgg13, vgg13_bn, vgg16, vgg16_bn, vgg19, vgg19_bn from torchvision.models.vgg import vgg11, vgg11_bn, vgg13, vgg13_bn, vgg16, vgg16_bn, vgg19, vgg19_bn
from torchvision.models.video import (
mc3_18,
mvit_v1_b,
mvit_v2_s,
r2plus1d_18,
r3d_18,
s3d,
swin3d_b,
swin3d_s,
swin3d_t,
)
from torchvision.models.vision_transformer import vit_b_16, vit_b_32, vit_h_14, vit_l_16, vit_l_32 from torchvision.models.vision_transformer import vit_b_16, vit_b_32, vit_h_14, vit_l_16, vit_l_32
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment