summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/vqvae/encoder.py
AgeCommit message (Collapse)Author
2021-08-04Add conv attention, up and downsampling to vqvae moduleGustaf Rydholm
2021-08-04New VQVAEGustaf Rydholm
2021-04-25Efficient net and non working transformer model.Gustaf Rydholm
2021-04-22Fixed training script, able to train vqvaeGustaf Rydholm
2021-04-07Fixed typing and typos, train script load config, reformattedGustaf Rydholm
2021-04-05Pre-commit fixes, optimizer loading fixGustaf Rydholm
2021-03-20Inital commit for refactoring to lightningGustaf Rydholm