Skip to content

GitLab

Unverified Commit b70e333a authored by Lutz Roeder's avatar Lutz Roeder Committed by GitHub
Browse files

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

parent 43e94b39
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
torchvision/models/resnet.py
View file @ b70e333a
...@@ -34,7 +34,6 @@ def conv1x1(in_planes, out_planes, stride=1): ...@@ -34,7 +34,6 @@ def conv1x1(in_planes, out_planes, stride=1):
class BasicBlock(nn.Module): class BasicBlock(nn.Module):
expansion = 1 expansion = 1
__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):
...@@ -75,7 +74,6 @@ class BasicBlock(nn.Module): ...@@ -75,7 +74,6 @@ 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):
......
torchvision/models/video/resnet.py
View file @ b70e333a
...@@ -81,7 +81,6 @@ class Conv3DNoTemporal(nn.Conv3d): ...@@ -81,7 +81,6 @@ class Conv3DNoTemporal(nn.Conv3d):
class BasicBlock(nn.Module): class BasicBlock(nn.Module):
__constants__ = ['downsample']
expansion = 1 expansion = 1
def __init__(self, inplanes, planes, conv_builder, stride=1, downsample=None): def __init__(self, inplanes, planes, conv_builder, stride=1, downsample=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