Skip to content

xcastilla/VisionTransformer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

[WIP] Visual Transformer

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!

How to Run

python3 src/main.py -c config/cifar-100.yaml

To do

  • Add requirements.txt, Dockerization, etc.
  • Linting and type checking.
  • Better modularization of data loading and training.
  • [...]

About

Simple Vision Transformer implementation using Keras

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages