From dbd33e4b6272cc2ad0d6aa1217d979ed956083de Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Wed, 22 Mar 2023 15:39:49 -0700 Subject: [PATCH] Update crop settings for new rexnet weights --- timm/models/rexnet.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/timm/models/rexnet.py b/timm/models/rexnet.py index f5c3ef67..4fa5c4ba 100644 --- a/timm/models/rexnet.py +++ b/timm/models/rexnet.py @@ -281,16 +281,18 @@ default_cfgs = generate_default_cfgs({ 'rexnetr_150.untrained': _cfg(), 'rexnetr_200.sw_in12k_ft_in1k': _cfg( hf_hub_id='timm/', - crop_pct=1.0, test_input_size=(3, 288, 288), license='apache-2.0'), + crop_pct=0.95, test_crop_pct=1.0, test_input_size=(3, 288, 288), license='apache-2.0'), 'rexnetr_300.sw_in12k_ft_in1k': _cfg( hf_hub_id='timm/', - crop_pct=1.0, test_input_size=(3, 288, 288), license='apache-2.0'), + crop_pct=0.95, test_crop_pct=1.0, test_input_size=(3, 288, 288), license='apache-2.0'), 'rexnetr_200.sw_in12k': _cfg( hf_hub_id='timm/', - num_classes=11821, crop_pct=1.0, license='apache-2.0'), + num_classes=11821, + crop_pct=0.95, test_crop_pct=1.0, test_input_size=(3, 288, 288), license='apache-2.0'), 'rexnetr_300.sw_in12k': _cfg( hf_hub_id='timm/', - num_classes=11821, crop_pct=1.0, license='apache-2.0'), + num_classes=11821, + crop_pct=0.95, test_crop_pct=1.0, test_input_size=(3, 288, 288), license='apache-2.0'), })