There is no bias in replaced maxpool so remove it

pull/554/head
Aman Arora 4 years ago
parent c7b40414bb
commit 1f45c6e926

@ -600,7 +600,7 @@ class ResNet(nn.Module):
self.maxpool = nn.MaxPool2d(kernel_size=3, stride=2, padding=1)
else:
self.maxpool = nn.Sequential(*[
nn.Conv2d(inplanes, inplanes, 3, stride=2, padding=1),
nn.Conv2d(inplanes, inplanes, 3, stride=2, padding=1, bias=False),
norm_layer(inplanes),
act_layer(inplace=True)
])

Loading…
Cancel
Save