Remove hardcoded values

4 years ago · 69f8c7123f
parent e7df94a760
commit 69f8c7123f
1 changed files with 2 additions and 2 deletions
--- a/timm/models/resnet.py
+++ b/timm/models/resnet.py
@ -601,8 +601,8 @@ class ResNet(nn.Module):
        else:
            self.maxpool = nn.Sequential(*[
                nn.Conv2d(inplanes, inplanes, 3, stride=2, padding=1), 
-                nn.BatchNorm2d(inplanes), 
-                nn.ReLU()
+                norm_layer(inplanes), 
+                act_layer(inplace=True)
            ])

        # Feature Blocks