From ac3470188b914c5d7a5058a7e28b9eb685a62427 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Wed, 14 Aug 2024 08:47:02 -0700 Subject: [PATCH] Add mobilenetv1_125 weights from ra4 / mnv4 hparam series --- timm/models/efficientnet.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py index 1ffb0d41..3aee0342 100644 --- a/timm/models/efficientnet.py +++ b/timm/models/efficientnet.py @@ -1248,7 +1248,11 @@ default_cfgs = generate_default_cfgs({ mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, test_input_size=(3, 256, 256), test_crop_pct=0.95, ), - 'mobilenetv1_125.untrained': _cfg(), + 'mobilenetv1_125.ra4_e3600_r224_in1k': _cfg( + hf_hub_id='timm/', + mean=IMAGENET_INCEPTION_MEAN, std=IMAGENET_INCEPTION_STD, + crop_pct=0.9, test_input_size=(3, 256, 256), test_crop_pct=1.0, + ), 'mobilenetv2_035.untrained': _cfg(), 'mobilenetv2_050.lamb_in1k': _cfg(