ps4vs / vision-transformer-pytorch-lightning Goto Github PK
View Code? Open in Web Editor NEWThis project forked from aniketrajpoot/vision-transformer-pytorch-lightning
Pytorch Lightning implementation of Vision Transformer with support for loading checkpoints saved in official Flax implementation.