Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.
tensorflow image-recognition self-attention vision-transformers imagenet-1k deeper-vits class-attention
-
Updated
Jun 9, 2023 - Jupyter Notebook