Keras
TF-Keras
File size: 572 Bytes
82c0ae2
 
 
 
90a9254
82c0ae2
90a9254
82c0ae2
90a9254
c96cc4b
90a9254
1
2
3
4
5
6
7
8
9
10
11
---
library_name: keras
---

This model is a TensorFlow port of ViT B-16 [1] trained with recipes from [2]. ImageNet-1k dataset was used for training purposes. You can refer to [this notebook](https://github.com/sayakpaul/probing-vits/blob/main/notebooks/load-jax-weights-vitb16.ipynb) to know how the porting was done.

## References

[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