Downloads · 30 days
0
0% of all-time downloads
probing-vits/vit_b16_patch16_224_i21k_i1k
vit_b16_patch16_224_i21k_i1k is a machine learning model from probing-vits. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for keras.
This model is a TensorFlow port of ViT B-16 [1] trained with recipes from [2]. It was first pre-trained on ImageNet-21k and was then fine-tuned on the ImageNet-1k dataset. You can refer to this notebook to know how th…
Downloads · 30 days
0
0% of all-time downloads
All-time downloads
3.7K
Public
Repo size
357 MB
Likes
1
Public
Click a slice to open those files.
.data-00000-of-00001346 MB · 98%
From the Hugging Face model README
This model is a TensorFlow port of ViT B-16 [1] trained with recipes from [2]. It was first pre-trained on ImageNet-21k and was then fine-tuned on the ImageNet-1k dataset. You can refer to this notebook to know how the porting was done.
[1] An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale: https://arxiv.org/abs/2010.11929
[2] How to train your ViT? Data, Augmentation, and Regularization in Vision Transformers: https://arxiv.org/abs/2106.10270