Skip to content

Commit a45b4bc

Browse files
committed
x and xx small edgenext models do benefit from larger test input size
1 parent a8e3405 commit a45b4bc

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

timm/models/edgenext.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,11 @@ def _cfg(url='', **kwargs):
3939

4040
default_cfgs = dict(
4141
edgenext_xx_small=_cfg(
42-
url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.0/edgenext_xx_small.pth"),
42+
url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.0/edgenext_xx_small.pth",
43+
test_input_size=(3, 288, 288), test_crop_pct=1.0),
4344
edgenext_x_small=_cfg(
44-
url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.0/edgenext_x_small.pth"),
45+
url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.0/edgenext_x_small.pth",
46+
test_input_size=(3, 288, 288), test_crop_pct=1.0),
4547
# edgenext_small=_cfg(
4648
# url="https://github.com/mmaaz60/EdgeNeXt/releases/download/v1.0/edgenext_small.pth"),
4749
edgenext_small=_cfg( # USI weights

0 commit comments

Comments
 (0)