summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/vqvae/encoder.py
AgeCommit message (Expand)Author
2021-11-21Remove VQVAE stuff, did not work...Gustaf Rydholm
2021-09-30Add num res blocks as a variableGustaf Rydholm
2021-09-19Add ability to set use norm in vqvaeGustaf Rydholm
2021-09-18Add more residual blocks in vqvaeGustaf Rydholm
2021-09-11Disable group norm in vqvae en/decoderGustaf Rydholm
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