"vscode:/vscode.git/clone" did not exist on "43346adc1ffa9051fc71be9af33fd982ee14c383"
Unverified Commit 035d99f1 authored by Aditya Oke's avatar Aditya Oke Committed by GitHub
Browse files

Add video models to torchhub (#7083)

parent 934ce3b8
......@@ -71,4 +71,15 @@ from torchvision.models.shufflenetv2 import (
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.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
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