Skip to content

Commit dbd33e4

Browse files
committed
Update crop settings for new rexnet weights
1 parent da6bdd4 commit dbd33e4

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

timm/models/rexnet.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -281,16 +281,18 @@ def _cfg(url='', **kwargs):
281281
'rexnetr_150.untrained': _cfg(),
282282
'rexnetr_200.sw_in12k_ft_in1k': _cfg(
283283
hf_hub_id='timm/',
284-
crop_pct=1.0, test_input_size=(3, 288, 288), license='apache-2.0'),
284+
crop_pct=0.95, test_crop_pct=1.0, test_input_size=(3, 288, 288), license='apache-2.0'),
285285
'rexnetr_300.sw_in12k_ft_in1k': _cfg(
286286
hf_hub_id='timm/',
287-
crop_pct=1.0, test_input_size=(3, 288, 288), license='apache-2.0'),
287+
crop_pct=0.95, test_crop_pct=1.0, test_input_size=(3, 288, 288), license='apache-2.0'),
288288
'rexnetr_200.sw_in12k': _cfg(
289289
hf_hub_id='timm/',
290-
num_classes=11821, crop_pct=1.0, license='apache-2.0'),
290+
num_classes=11821,
291+
crop_pct=0.95, test_crop_pct=1.0, test_input_size=(3, 288, 288), license='apache-2.0'),
291292
'rexnetr_300.sw_in12k': _cfg(
292293
hf_hub_id='timm/',
293-
num_classes=11821, crop_pct=1.0, license='apache-2.0'),
294+
num_classes=11821,
295+
crop_pct=0.95, test_crop_pct=1.0, test_input_size=(3, 288, 288), license='apache-2.0'),
294296
})
295297

296298

0 commit comments

Comments
 (0)