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', }