mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update arxiv link

pull/1007/head
qwertyforce 2021-11-29 21:41:00 +03:00 committed by GitHub
parent f7d210d759
commit ccb3815360
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/models/vision_transformer_hybrid.py
View File

@ -6,7 +6,7 @@ A PyTorch implement of the Hybrid Vision Transformers as described in:
- https://arxiv.org/abs/2010.11929 - https://arxiv.org/abs/2010.11929
`How to train your ViT? Data, Augmentation, and Regularization in Vision Transformers` `How to train your ViT? Data, Augmentation, and Regularization in Vision Transformers`
- https://arxiv.org/abs/2106.TODO - https://arxiv.org/abs/2106.10270
NOTE These hybrid model definitions depend on code in vision_transformer.py. NOTE These hybrid model definitions depend on code in vision_transformer.py.
They were moved here to keep file sizes sane. They were moved here to keep file sizes sane.
@ -360,4 +360,4 @@ def vit_base_resnet50d_224(pretrained=False, **kwargs):
model_kwargs = dict(embed_dim=768, depth=12, num_heads=12, **kwargs) model_kwargs = dict(embed_dim=768, depth=12, num_heads=12, **kwargs)
model = _create_vision_transformer_hybrid( model = _create_vision_transformer_hybrid(
'vit_base_resnet50d_224', backbone=backbone, pretrained=pretrained, **model_kwargs) 'vit_base_resnet50d_224', backbone=backbone, pretrained=pretrained, **model_kwargs)
return model return model