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

Add link to model weights on Hugging Face

Browse Source
This commit is contained in:
Beckschen 2024-05-17 06:51:35 -04:00
parent 530fb49e7e
commit 7a2ad6bce1
1 changed files with 2 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
timm/models/vitamin.py
View File

@ -1,7 +1,7 @@
""" ViTamin
Paper: Designing Scalable Vison Models in the Vision-Language Era
Model Weights on Huggingface: https://huggingface.co/collections/jienengchen/vitamin-family-661048126b72debdaca060bf
A family of model weights on Huggingface: https://huggingface.co/collections/jienengchen/vitamin-family-661048126b72debdaca060bf
@inproceedings{chen2024vitamin,
title={ViTamin: Designing Scalable Vision Models in the Vision-language Era},
@ -530,10 +530,3 @@ def vitamin_xlarge_384(pretrained=False, **kwargs) -> VisionTransformer:
model = _create_vision_transformer_hybrid(
'vitamin_xlarge_384', backbone=backbone, pretrained=pretrained, **dict(model_args, **kwargs))
return model
if __name__ == "__main__":
model = timm.create_model('vitamin_large', num_classes=10).cuda()
x = torch.rand([2,3,224,224]).cuda()
y = model(x)
print(y.shape)