From 2bb65bd8750755112da11124c2fdc4895bd971a8 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 1 Apr 2021 20:00:41 -0700 Subject: [PATCH] Wrong default_cfg pool_size for L1 --- timm/models/nfnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/nfnet.py b/timm/models/nfnet.py index 537fb15d..1fa0d212 100644 --- a/timm/models/nfnet.py +++ b/timm/models/nfnet.py @@ -109,7 +109,7 @@ default_cfgs = dict( pool_size=(7, 7), input_size=(3, 224, 224), test_input_size=(3, 288, 288), crop_pct=1.0), eca_nfnet_l1=_dcfg( url='', - pool_size=(7, 7), input_size=(3, 256, 256), test_input_size=(3, 320, 320), crop_pct=1.0), + pool_size=(8, 8), input_size=(3, 256, 256), test_input_size=(3, 320, 320), crop_pct=1.0), nf_regnet_b0=_dcfg( url='', pool_size=(6, 6), input_size=(3, 192, 192), test_input_size=(3, 256, 256), first_conv='stem.conv'),