Remove hardcoded values

pull/554/head
Aman Arora 4 years ago
parent e7df94a760
commit 69f8c7123f

@ -601,8 +601,8 @@ class ResNet(nn.Module):
else: else:
self.maxpool = nn.Sequential(*[ self.maxpool = nn.Sequential(*[
nn.Conv2d(inplanes, inplanes, 3, stride=2, padding=1), nn.Conv2d(inplanes, inplanes, 3, stride=2, padding=1),
nn.BatchNorm2d(inplanes), norm_layer(inplanes),
nn.ReLU() act_layer(inplace=True)
]) ])
# Feature Blocks # Feature Blocks

Loading…
Cancel
Save