From d76530582164740e65b6992148d2a755f16cde6b Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 2 Jul 2022 15:56:17 -0700 Subject: [PATCH] Remove first_conv for resnetaa50 def --- timm/models/resnet.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/timm/models/resnet.py b/timm/models/resnet.py index 28f3cdba..e5a6b791 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -274,8 +274,7 @@ default_cfgs = { interpolation='bicubic', first_conv='conv1.0'), 'resnetaa50': _cfg( url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-rsb-weights/resnetaa50_a1h-4cf422b3.pth', - test_input_size=(3, 288, 288), test_crop_pct=1.0, - interpolation='bicubic', first_conv='conv1.0'), + test_input_size=(3, 288, 288), test_crop_pct=1.0, interpolation='bicubic'), 'resnetaa50d': _cfg( url='', interpolation='bicubic', first_conv='conv1.0'),