Skip to content

GitLab

Commit 3c1ab2c1 authored by eellison's avatar eellison Committed by Francisco Massa
Browse files

make resnext scriptable (#1343)

parent e8b830fc
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
test/test_models.py
View file @ 3c1ab2c1
...@@ -29,7 +29,7 @@ def get_available_video_models(): ...@@ -29,7 +29,7 @@ def get_available_video_models():
torchub_models = { torchub_models = {
"deeplabv3_resnet101": False, "deeplabv3_resnet101": False,
"mobilenet_v2": True, "mobilenet_v2": True,
"resnext50_32x4d": False, "resnext50_32x4d": True,
"fcn_resnet101": False, "fcn_resnet101": False,
"googlenet": False, "googlenet": False,
"densenet121": False, "densenet121": False,
......
torchvision/models/resnet.py
View file @ 3c1ab2c1
...@@ -75,6 +75,7 @@ class BasicBlock(nn.Module): ...@@ -75,6 +75,7 @@ class BasicBlock(nn.Module):
class Bottleneck(nn.Module): class Bottleneck(nn.Module):
expansion = 4 expansion = 4
__constants__ = ['downsample']
def __init__(self, inplanes, planes, stride=1, downsample=None, groups=1, def __init__(self, inplanes, planes, stride=1, downsample=None, groups=1,
base_width=64, dilation=1, norm_layer=None): base_width=64, dilation=1, norm_layer=None):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment