Add some 384x384 small model weights, 3 variants of mnv4 conv medium …

…on in12k pretrain, and resnetv2-34d on in1k
huggingface · Nov 17, 2024 · b7a4b49 · b7a4b49
1 parent facae65
commit b7a4b49
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/timm/models/mobilenetv3.py b/timm/models/mobilenetv3.py
@@ -1031,6 +1031,23 @@ def _cfg(url: str = '', **kwargs):
     'mobilenetv4_conv_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_conv_medium.e180_r384_in12k': _cfg(
+        hf_hub_id='timm/',
+        num_classes=11821,
+        input_size=(3, 384, 384), pool_size=(12, 12),
+        crop_pct=1.0, interpolation='bicubic'),
+    'mobilenetv4_conv_medium.e180_ad_r384_in12k': _cfg(
+        hf_hub_id='timm/',
+        num_classes=11821,
+        input_size=(3, 384, 384), pool_size=(12, 12),
+        crop_pct=1.0, interpolation='bicubic'),
+    'mobilenetv4_conv_medium.e250_r384_in12k': _cfg(
+        hf_hub_id='timm/',
+        num_classes=11821,
+        input_size=(3, 384, 384), pool_size=(12, 12),
+        crop_pct=1.0, interpolation='bicubic'),
+
     'mobilenetv4_conv_large.e600_r384_in1k': _cfg(
         hf_hub_id='timm/',
         input_size=(3, 384, 384), pool_size=(12, 12),

diff --git a/timm/models/resnetv2.py b/timm/models/resnetv2.py
@@ -710,6 +710,10 @@ def _cfg(url='', **kwargs):
         hf_hub_id='timm/',
         interpolation='bicubic', crop_pct=0.9, test_input_size=(3, 288, 288), test_crop_pct=1.0,
         first_conv='stem.conv1'),
+    'resnetv2_34d.ra4_e3600_r384_in1k': _cfg(
+        hf_hub_id='timm/',
+        interpolation='bicubic', crop_pct=1.0, test_input_size=(3, 448, 448), pool_size=(12, 12),
+        first_conv='stem.conv1'),
     'resnetv2_50.a1h_in1k': _cfg(
         hf_hub_id='timm/',
         interpolation='bicubic', crop_pct=0.95, test_input_size=(3, 288, 288), test_crop_pct=1.0),