x and xx small edgenext models do benefit from larger test input size

pull/1345/head
Ross Wightman 2 years ago
parent a8e34051c1
commit a45b4bce9a

@ -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

Loading…
Cancel
Save