Upload video models (#1208)

pytorch · Aug 6, 2019 · d31eafa · d31eafa
1 parent 18577d0
commit d31eafa
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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',
 }