summaryrefslogtreecommitdiff
path: root/text_recognizer/models/vqvae.py
AgeCommit message (Collapse)Author
2021-09-11Rename vq loss to commitment lossGustaf Rydholm
2021-08-08Add VQGAN and loss functionGustaf Rydholm
2021-08-06Add logging of acc in VQVAE modelGustaf Rydholm
2021-08-06Remove lr args from model, add Cosine lr, fix to vqvae stackGustaf Rydholm
2021-08-04Add char htr experiment, rename from ocr to htr, vqvae loss collapsesGustaf Rydholm
2021-08-04New VQVAEGustaf Rydholm
2021-08-02Fix log import, fix mapping in datamodules, fix nn modules can be hashedGustaf Rydholm
2021-07-30attr bug fix, properly loading networkGustaf Rydholm
2021-07-06Refactoring data with attrs and refactor conf for hydraGustaf Rydholm
2021-07-05Refactor with attr, working on cnn+transformer networkGustaf Rydholm
2021-06-27Remove unused importsGustaf Rydholm
2021-05-04Nyströmer implemented but not testedGustaf Rydholm
2021-05-02Attention layer finishedGustaf Rydholm
2021-04-22Fixed training script, able to train vqvaeGustaf Rydholm