mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix in_chans switching on create

Browse Source
This commit is contained in:
Ross Wightman 2024-06-06 17:56:14 -07:00
parent fc1b66a51d
commit ad026e6e33
1 changed files with 3 additions and 2 deletions

5
timm/models/vision_transformer_hybrid.py

@ -389,12 +389,12 @@ default_cfgs = generate_default_cfgs({
'vit_base_mci_224.apple_mclip': _cfg( 'vit_base_mci_224.apple_mclip': _cfg(
url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_b.pt', url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_b.pt',
num_classes=512, num_classes=512,
mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.0.conv.weight', mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.0.conv',
), ),
'vit_base_mci_224.apple_mclip_lt': _cfg( 'vit_base_mci_224.apple_mclip_lt': _cfg(
url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_blt.pt', url='https://docs-assets.developer.apple.com/ml-research/datasets/mobileclip/mobileclip_blt.pt',
num_classes=512, num_classes=512,
mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.0.conv.weight', mean=(0., 0., 0.), std=(1., 1., 1.), first_conv='patch_embed.backbone.0.conv',
), ),
}) })
@ -552,6 +552,7 @@ def vit_base_mci_224(pretrained=False, **kwargs) -> VisionTransformer:
stride=(4, 2, 2), stride=(4, 2, 2),
kernel_size=(4, 2, 2), kernel_size=(4, 2, 2),
padding=0, padding=0,
in_chans=kwargs.get('in_chans', 3),
act_layer=nn.GELU, act_layer=nn.GELU,
) )
model_args = dict(embed_dim=768, depth=12, num_heads=12, no_embed_class=True) model_args = dict(embed_dim=768, depth=12, num_heads=12, no_embed_class=True)