Wrong default_cfg pool_size for L1

4 years ago · 2bb65bd875
parent bf2ca6bdf4
commit 2bb65bd875
1 changed files with 1 additions and 1 deletions
--- 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'),