From a1cb25066e26c8bf8fa410987b808899657c9e20 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 7 Jul 2022 22:02:57 -0700 Subject: [PATCH] Add edgnext_small_rw weights trained with swin like recipe. Better than original 'small' but not the recent 'USI' distilled weights. --- timm/models/edgenext.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/timm/models/edgenext.py b/timm/models/edgenext.py index 97971ba6..29316b9a 100644 --- a/timm/models/edgenext.py +++ b/timm/models/edgenext.py @@ -46,10 +46,13 @@ default_cfgs = dict( # url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.0/edgenext_small.pth"), edgenext_small=_cfg( # USI weights url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.1/edgenext_small_usi.pth", - crop_pct=0.95 + crop_pct=0.95, test_input_size=(3, 320, 320), test_crop_pct=1.0, ), - edgenext_small_rw=_cfg(), + edgenext_small_rw=_cfg( + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-attn-weights/edgenext_small_rw-sw-b00041bb.pth', + test_input_size=(3, 320, 320), test_crop_pct=1.0, + ), )