Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

84_0

Browse Source
pull/595/head
talrid 4 years ago
parent 0968bdeca3
commit cf0e371594
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
timm/models/vision_transformer.py
Unescape View File

@ -126,7 +126,7 @@ default_cfgs = {
), ),
'vit_base_patch16_224_1k_miil': _cfg( 'vit_base_patch16_224_1k_miil': _cfg(
url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/ImageNet_21K_P/models/timm' url='https://miil-public-eu.oss-eu-central-1.aliyuncs.com/model-zoo/ImageNet_21K_P/models/timm'
'/vit_base_patch16_224_1k_miil_84_4.pth', '/vit_base_patch16_224_1k_miil_84_0.pth',
mean=(0, 0, 0), std=(1, 1, 1), crop_pct=0.875, interpolation='bilinear', mean=(0, 0, 0), std=(1, 1, 1), crop_pct=0.875, interpolation='bilinear',
), ),
} }
@ -714,5 +714,5 @@ def vit_base_patch16_224_1k_miil(pretrained=False, **kwargs):
Weights taken from: https://github.com/Alibaba-MIIL/ImageNet21K Weights taken from: https://github.com/Alibaba-MIIL/ImageNet21K
""" """
model_kwargs = dict(patch_size=16, embed_dim=768, depth=12, num_heads=12, qkv_bias=False, **kwargs) model_kwargs = dict(patch_size=16, embed_dim=768, depth=12, num_heads=12, qkv_bias=False, **kwargs)
model = _create_vision_transformer('vit_base_patch16_224_1k_miil_84_4', pretrained=pretrained, **model_kwargs) model = _create_vision_transformer('vit_base_patch16_224_1k_miil', pretrained=pretrained, **model_kwargs)
return model return model
Write Preview
Loading…
Cancel
Save