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

Summary: Pull Request resolved: #2419 Reviewed By: zhangguanheng66 Differential Revision: D22432605 Pulled By: fmassa fbshipit-source-id: f677348d87fb7ee852d158222387313fd26d5dc3
pytorch · Jul 8, 2020 · 2eddf2a · 2eddf2a
1 parent 70503fa
commit 2eddf2a
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/torchvision/models/resnet.py b/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):

diff --git a/torchvision/models/video/resnet.py b/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):