Toy implementation of An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale.
The code is not yet finalized, linted nor well tested, so use at your own risk!
python3 src/main.py -c config/cifar-100.yaml- Add requirements.txt, Dockerization, etc.
- Linting and type checking.
- Better modularization of data loading and training.
- [...]