From d31eafa9c59cd59533dd0b89a36d371e55337a48 Mon Sep 17 00:00:00 2001 From: Francisco Massa Date: Tue, 6 Aug 2019 16:15:22 -0700 Subject: [PATCH] Upload video models (#1208) --- torchvision/models/video/resnet.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/torchvision/models/video/resnet.py b/torchvision/models/video/resnet.py index 8fe3bd1007f..a9e59a149c0 100644 --- a/torchvision/models/video/resnet.py +++ b/torchvision/models/video/resnet.py @@ -7,8 +7,9 @@ __all__ = ['r3d_18', 'mc3_18', 'r2plus1d_18'] model_urls = { - 'resnet18': 'https://download.pytorch.org/models/', - 'resnet34': 'https://download.pytorch.org/models/', + 'r3d_18': 'https://download.pytorch.org/models/r3d_18-b3b3357e.pth', + 'mc3_18': 'https://download.pytorch.org/models/mc3_18-a90a0ba3.pth', + 'r2plus1d_18': 'https://download.pytorch.org/models/r2plus1d_18-91a641e6.pth', }