Add mobilenetv1_125 weights from ra4 / mnv4 hparam series

huggingface · Aug 14, 2024 · ac34701 · ac34701
1 parent 907555f
commit ac34701
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py
@@ -1248,7 +1248,11 @@ def _cfg(url='', **kwargs):
         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(