/cait-tf

Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Stargazers