From fc1b66a51d35201ecc13c1c5697c4b2e28c9c49d Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 6 Jun 2024 13:42:26 -0700 Subject: [PATCH] Fix first conv name for mci vit-b --- timm/models/vision_transformer_hybrid.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/vision_transformer_hybrid.py b/timm/models/vision_transformer_hybrid.py index 3501565c..c16e7c78 100644 --- a/timm/models/vision_transformer_hybrid.py +++ b/timm/models/vision_transformer_hybrid.py @@ -389,12 +389,12 @@ default_cfgs = generate_default_cfgs({ 'vit_base_mci_224.apple_mclip': _cfg( url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_b.pt', num_classes=512, - mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.conv1.0', + mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.0.conv.weight', ), 'vit_base_mci_224.apple_mclip_lt': _cfg( url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_blt.pt', num_classes=512, - mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.conv1.0', + mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.0.conv.weight', ), })