Remove 'downsample' constants (#1721) (#1923)

Author: lutzroeder

torchvision/models/resnet.py

@@ -34,7 +34,6 @@ def conv1x1(in_planes, out_planes, stride=1):
 
 class BasicBlock(nn.Module):
     expansion = 1
-    __constants__ = ['downsample']
 
     def __init__(self, inplanes, planes, stride=1, downsample=None, groups=1,
                  base_width=64, dilation=1, norm_layer=None):
@@ -75,7 +74,6 @@ def forward(self, x):
 
 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):

torchvision/models/video/resnet.py

@@ -81,7 +81,6 @@ def get_downsample_stride(stride):
 
 class BasicBlock(nn.Module):
 
-    __constants__ = ['downsample']
     expansion = 1
 
     def __init__(self, inplanes, planes, conv_builder, stride=1, downsample=None):