summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/vqvae/decoder.py
AgeCommit message (Expand)Author
2021-08-08Add VQGAN and loss functionGustaf Rydholm
2021-08-06Fix VQVAE into en/decoder, bug in wandb artifact code uploadingGustaf Rydholm
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