Skip to content

Commit a1cb250

Browse files
committed
Add edgnext_small_rw weights trained with swin like recipe. Better than original 'small' but not the recent 'USI' distilled weights.
1 parent 7c7ecd2 commit a1cb250

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

timm/models/edgenext.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,10 +46,13 @@ def _cfg(url='', **kwargs):
4646
# url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.0/edgenext_small.pth"),
4747
edgenext_small=_cfg( # USI weights
4848
url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.1/edgenext_small_usi.pth",
49-
crop_pct=0.95
49+
crop_pct=0.95, test_input_size=(3, 320, 320), test_crop_pct=1.0,
5050
),
5151

52-
edgenext_small_rw=_cfg(),
52+
edgenext_small_rw=_cfg(
53+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-attn-weights/edgenext_small_rw-sw-b00041bb.pth',
54+
test_input_size=(3, 320, 320), test_crop_pct=1.0,
55+
),
5356
)
5457

5558

0 commit comments

Comments
 (0)