Commit 867d5a50 authored by yl-1993's avatar yl-1993
Browse files

consistent param list of block

parent e363982a
......@@ -28,7 +28,8 @@ class BasicBlock(nn.Module):
stride=1,
dilation=1,
downsample=None,
style='pytorch'):
style='pytorch',
with_cp=False):
super(BasicBlock, self).__init__()
self.conv1 = conv3x3(inplanes, planes, stride, dilation)
self.bn1 = nn.BatchNorm2d(planes)
......@@ -38,6 +39,7 @@ class BasicBlock(nn.Module):
self.downsample = downsample
self.stride = stride
self.dilation = dilation
assert not with_cp
def forward(self, x):
residual = x
......
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