From 69f8c7123ffec03adc23262dd85cc55a9670204d Mon Sep 17 00:00:00 2001 From: Aman Arora Date: Sun, 2 May 2021 01:40:06 +0000 Subject: [PATCH] Remove hardcoded values --- timm/models/resnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/resnet.py b/timm/models/resnet.py index fe10ff22..24d1bf3b 100644 --- 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