Skip to content

Commit fc1b66a

Browse files
committed
Fix first conv name for mci vit-b
1 parent 88a1006 commit fc1b66a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/models/vision_transformer_hybrid.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -389,12 +389,12 @@ def _cfg(url='', **kwargs):
389389
'vit_base_mci_224.apple_mclip': _cfg(
390390
url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_b.pt',
391391
num_classes=512,
392-
mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.conv1.0',
392+
mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.0.conv.weight',
393393
),
394394
'vit_base_mci_224.apple_mclip_lt': _cfg(
395395
url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_blt.pt',
396396
num_classes=512,
397-
mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.conv1.0',
397+
mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.0.conv.weight',
398398
),
399399
})
400400

0 commit comments

Comments
 (0)