summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/vqvae
AgeCommit message (Expand)Author
2021-09-19Add ability to set use norm in vqvaeGustaf Rydholm
2021-09-18Add comment for quantizationGustaf Rydholm
2021-09-18Add more residual blocks in vqvaeGustaf Rydholm
2021-09-11Disable group norm in vqvae en/decoderGustaf Rydholm
2021-09-11Rename vq loss to commitment lossGustaf 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-04Add char htr experiment, rename from ocr to htr, vqvae loss collapsesGustaf Rydholm
2021-08-04New VQVAEGustaf Rydholm
2021-07-30attr bug fix, properly loading networkGustaf 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