Skip to content

Comments

add missing ViTamin configs#978

Merged
rwightman merged 1 commit intomlfoundations:mainfrom
xywei00:xwei-vitamin-configs
Nov 24, 2024
Merged

add missing ViTamin configs#978
rwightman merged 1 commit intomlfoundations:mainfrom
xywei00:xwei-vitamin-configs

Conversation

@xywei00
Copy link
Contributor

@xywei00 xywei00 commented Nov 5, 2024

In src/open_clip/model_configs, two configs of the ViTamin series are missing: ViTamin-L-384 and ViTamin-L2-384.

The two added models are evaluated (with pretrained weights) on the Datacomp benchmark, where ViTamin-L-384 has 81.83% ImageNet top 1 and 67.12% Average, and ViTamin-L2-384 has 82.05% ImageNet top 1 and 67.97% Average, matching the performance reported in the paper.

@rwightman rwightman merged commit ea524b9 into mlfoundations:main Nov 24, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants