├── .gitignore ├── README.md ├── configs ├── transformer_cat.yml ├── transformer_dmol.yml └── transformer_tiny.yml ├── image_transformer.py ├── requirements.txt └── train_transformer.py /.gitignore: -------------------------------------------------------------------------------- 1 | transformer_logs/ 2 | venv/ 3 | datasets/ 4 | __pycache__ 5 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sahajgarg/image_transformer/HEAD/README.md -------------------------------------------------------------------------------- /configs/transformer_cat.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sahajgarg/image_transformer/HEAD/configs/transformer_cat.yml -------------------------------------------------------------------------------- /configs/transformer_dmol.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sahajgarg/image_transformer/HEAD/configs/transformer_dmol.yml -------------------------------------------------------------------------------- /configs/transformer_tiny.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sahajgarg/image_transformer/HEAD/configs/transformer_tiny.yml -------------------------------------------------------------------------------- /image_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sahajgarg/image_transformer/HEAD/image_transformer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sahajgarg/image_transformer/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sahajgarg/image_transformer/HEAD/train_transformer.py --------------------------------------------------------------------------------