make resnext scriptable (#1343)

3c1ab2c1 · eellison · Francisco Massa · e8b830fc · 3c1ab2c1 · 3c1ab2c1
Commit 3c1ab2c1 authored Sep 17, 2019 by eellison Committed by Francisco Massa Sep 17, 2019
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

test/test_models.py test/test_models.py +1 -1

torchvision/models/resnet.py torchvision/models/resnet.py +1 -0

No files found.
--- a/test/test_models.py
+++ b/test/test_models.py
@@ -29,7 +29,7 @@ def get_available_video_models():
 torchub_models = {
    "deeplabv3_resnet101": False,
    "mobilenet_v2": True,
-    "resnext50_32x4d": False,
+    "resnext50_32x4d": True,
    "fcn_resnet101": False,
    "googlenet": False,
    "densenet121": False,

--- a/torchvision/models/resnet.py
+++ b/torchvision/models/resnet.py
@@ -75,6 +75,7 @@ class BasicBlock(nn.Module):

 class Bottleneck(nn.Module):
    expansion = 4
+    __constants__ = ['downsample']

    def __init__(self, inplanes, planes, stride=1, downsample=None, groups=1,
                 base_width=64, dilation=1, norm_layer=None):