summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/vq_transformer.py
AgeCommit message (Collapse)Author
2022-06-27Add vq transformer networkGustaf Rydholm
2021-11-21Remove VQVAE stuff, did not work...Gustaf Rydholm
2021-11-17Update VqTransformerGustaf Rydholm
2021-10-10Remove importsGustaf Rydholm
2021-09-30Bug fix for loading pretrained vq encoderGustaf Rydholm
2021-09-19Add loading of encoder in vq transformer networkGustaf Rydholm
2021-09-18Add Vq transfomer modelGustaf Rydholm
2021-08-06Fix VQVAE into en/decoder, bug in wandb artifact code uploadingGustaf Rydholm
2021-08-04New VQVAEGustaf Rydholm
2021-04-05Add new training loop with PyTorch Lightning, remove stale filesGustaf Rydholm
2021-03-20Inital commit for refactoring to lightningGustaf Rydholm