From a45b4bce9a022d413eb27a342a7a9997580bb4aa Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 8 Jul 2022 10:53:27 -0700 Subject: [PATCH] x and xx small edgenext models do benefit from larger test input size --- timm/models/edgenext.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/timm/models/edgenext.py b/timm/models/edgenext.py index 29316b9a..a81bd9b0 100644 --- a/timm/models/edgenext.py +++ b/timm/models/edgenext.py @@ -39,9 +39,11 @@ def _cfg(url='', **kwargs): default_cfgs = dict( edgenext_xx_small=_cfg( - url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.0/edgenext_xx_small.pth"), + url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.0/edgenext_xx_small.pth", + test_input_size=(3, 288, 288), test_crop_pct=1.0), edgenext_x_small=_cfg( - url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.0/edgenext_x_small.pth"), + url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.0/edgenext_x_small.pth", + test_input_size=(3, 288, 288), test_crop_pct=1.0), # edgenext_small=_cfg( # url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.0/edgenext_small.pth"), edgenext_small=_cfg( # USI weights