Add mobilenetv4 hybrid-large weights

huggingface · Jun 17, 2024 · 16e082e · 16e082e
1 parent e41125c
commit 16e082e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py
@@ -1024,12 +1024,13 @@ def _cfg(url: str = '', **kwargs):
     'mobilenetv4_hybrid_medium.e500_r224_in1k': _cfg(
         hf_hub_id='timm/',
         crop_pct=0.95, test_input_size=(3, 256, 256), test_crop_pct=1.0, interpolation='bicubic'),
+    'mobilenetv4_hybrid_large.e600_r384_in1k': _cfg(
+        hf_hub_id='timm/',
+        input_size=(3, 384, 384), pool_size=(12, 12),
+        crop_pct=0.95, test_input_size=(3, 448, 448), test_crop_pct=1.0, interpolation='bicubic'),
     'mobilenetv4_hybrid_large.r256': _cfg(
         # hf_hub_id='timm/',
         input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=0.95, interpolation='bicubic'),
-    'mobilenetv4_hybrid_large.r384': _cfg(
-        # hf_hub_id='timm/',
-        input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=0.95, interpolation='bicubic'),
 
     # experimental
     'mobilenetv4_conv_aa_medium.untrained': _cfg(