diff --git a/timm/models/byobnet.py b/timm/models/byobnet.py index 7464b901..a504b726 100644 --- a/timm/models/byobnet.py +++ b/timm/models/byobnet.py @@ -1919,22 +1919,27 @@ default_cfgs = generate_default_cfgs({ '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'), ), }) diff --git a/timm/models/fastvit.py b/timm/models/fastvit.py index 40049fae..e175c738 100644 --- a/timm/models/fastvit.py +++ b/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, }