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
Hide 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):
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 @@ 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):
......
torchvision/models/video/resnet.py
View file @ b70e333a
......@@ -81,7 +81,6 @@ class Conv3DNoTemporal(nn.Conv3d):
class BasicBlock(nn.Module):
__constants__ = ['downsample']
expansion = 1
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