Fix first_conv for mobileone and fastvit

Author: rwightman

timm/models/byobnet.py

@@ -1919,22 +1919,27 @@ def _cfgr(url='', **kwargs):
     'mobileone_s0.apple_in1k': _cfg(
         hf_hub_id='timm/',
         crop_pct=0.875,
+        first_conv=('stem.conv_kxk.0.conv', 'stem.conv_scale.conv'),
     ),
     'mobileone_s1.apple_in1k': _cfg(
         hf_hub_id='timm/',
         crop_pct=0.9,
+        first_conv=('stem.conv_kxk.0.conv', 'stem.conv_scale.conv'),
     ),
     'mobileone_s2.apple_in1k': _cfg(
         hf_hub_id='timm/',
         crop_pct=0.9,
+        first_conv=('stem.conv_kxk.0.conv', 'stem.conv_scale.conv'),
     ),
     'mobileone_s3.apple_in1k': _cfg(
         hf_hub_id='timm/',
         crop_pct=0.9,
+        first_conv=('stem.conv_kxk.0.conv', 'stem.conv_scale.conv'),
     ),
     'mobileone_s4.apple_in1k': _cfg(
         hf_hub_id='timm/',
         crop_pct=0.9,
+        first_conv=('stem.conv_kxk.0.conv', 'stem.conv_scale.conv'),
     ),
 })
 

timm/models/fastvit.py

@@ -1264,7 +1264,7 @@ def _cfg(url="", **kwargs):
         "interpolation": "bicubic",
         "mean": IMAGENET_DEFAULT_MEAN,
         "std": IMAGENET_DEFAULT_STD,
-        'first_conv': 'stem.0.conv_kxk.0.conv',
+        'first_conv': ('stem.0.conv_kxk.0.conv', 'stem.0.conv_scale.conv'),
         "classifier": "head.fc",
         **kwargs,
     }