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