Upload video models (#1208)

d31eafa9 · Francisco Massa · GitHub · 18577d02 · d31eafa9
Unverified Commit d31eafa9 authored Aug 06, 2019 by Francisco Massa Committed by GitHub Aug 06, 2019
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

torchvision/models/video/resnet.py torchvision/models/video/resnet.py +3 -2

No files found.
--- a/torchvision/models/video/resnet.py
+++ b/torchvision/models/video/resnet.py
@@ -7,8 +7,9 @@ from ..utils import load_state_dict_from_url
 __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',
 }