├── .gitignore ├── LICENSE ├── README.md ├── assets ├── butterfly_cropped.png ├── class_attention_heads_0.png ├── class_attention_heads_1.png ├── cls_attn_heads.png ├── cls_saliency.png └── saliency_attention_map.png ├── cait ├── __init__.py ├── layers │ ├── __init__.py │ ├── ca.py │ ├── ls.py │ ├── sd.py │ └── tha.py ├── model_configs │ ├── __init__.py │ └── base_config.py └── models.py ├── convert.py ├── convert_all_models.py ├── hub_utilities ├── README.md ├── export_for_hub.py └── generate_doc.py ├── i1k_eval ├── README.md ├── eval-cait.ipynb └── imagenet_class_index.json ├── notebooks ├── classification.ipynb └── finetune.ipynb ├── requirements.txt └── utils ├── __init__.py └── helpers.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/README.md -------------------------------------------------------------------------------- /assets/butterfly_cropped.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/assets/butterfly_cropped.png -------------------------------------------------------------------------------- /assets/class_attention_heads_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/assets/class_attention_heads_0.png -------------------------------------------------------------------------------- /assets/class_attention_heads_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/assets/class_attention_heads_1.png -------------------------------------------------------------------------------- /assets/cls_attn_heads.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/assets/cls_attn_heads.png -------------------------------------------------------------------------------- /assets/cls_saliency.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/assets/cls_saliency.png -------------------------------------------------------------------------------- /assets/saliency_attention_map.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/assets/saliency_attention_map.png -------------------------------------------------------------------------------- /cait/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cait/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/cait/layers/__init__.py -------------------------------------------------------------------------------- /cait/layers/ca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/cait/layers/ca.py -------------------------------------------------------------------------------- /cait/layers/ls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/cait/layers/ls.py -------------------------------------------------------------------------------- /cait/layers/sd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/cait/layers/sd.py -------------------------------------------------------------------------------- /cait/layers/tha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/cait/layers/tha.py -------------------------------------------------------------------------------- /cait/model_configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cait/model_configs/base_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/cait/model_configs/base_config.py -------------------------------------------------------------------------------- /cait/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/cait/models.py -------------------------------------------------------------------------------- /convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/convert.py -------------------------------------------------------------------------------- /convert_all_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/convert_all_models.py -------------------------------------------------------------------------------- /hub_utilities/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/hub_utilities/README.md -------------------------------------------------------------------------------- /hub_utilities/export_for_hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/hub_utilities/export_for_hub.py -------------------------------------------------------------------------------- /hub_utilities/generate_doc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/hub_utilities/generate_doc.py -------------------------------------------------------------------------------- /i1k_eval/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/i1k_eval/README.md -------------------------------------------------------------------------------- /i1k_eval/eval-cait.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/i1k_eval/eval-cait.ipynb -------------------------------------------------------------------------------- /i1k_eval/imagenet_class_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/i1k_eval/imagenet_class_index.json -------------------------------------------------------------------------------- /notebooks/classification.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/notebooks/classification.ipynb -------------------------------------------------------------------------------- /notebooks/finetune.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/notebooks/finetune.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/cait-tf/HEAD/utils/helpers.py --------------------------------------------------------------------------------