summaryrefslogtreecommitdiff
path: root/text_recognizer/models
AgeCommit message (Expand)Author
2021-09-30Bug fix transformer dim, comment CER/WER metricsGustaf Rydholm
2021-09-19Linting of Base modelGustaf Rydholm
2021-09-19Add weight for commitment loss in vq transformerGustaf Rydholm
2021-09-18Add Vq transfomer modelGustaf Rydholm
2021-09-11Rename vq loss to commitment lossGustaf Rydholm
2021-08-30Bug fix when loading no lr schedulerGustaf Rydholm
2021-08-29Remove uploading of code to Wandb, upload config insteadGustaf Rydholm
2021-08-15Updates to VQGAN lossGustaf Rydholm
2021-08-10Update to config and logging in VQGANGustaf Rydholm
2021-08-08Add training of VQGANGustaf 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-03Training working, multiple bug fixesGustaf Rydholm
2021-08-02Fix log import, fix mapping in datamodules, fix nn modules can be hashedGustaf Rydholm
2021-07-30black reformattingGustaf Rydholm
2021-07-30attr bug fix, properly loading networkGustaf Rydholm
2021-07-30Black formattingGustaf Rydholm
2021-07-29Configs, refactor with attrs, fix attr bug in iamGustaf Rydholm
2021-07-23Remove dinoGustaf 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-07-05Add new updates to model and networks, reimplementing with attrGustaf Rydholm
2021-06-27Remove unused importsGustaf Rydholm
2021-06-26UpdatesGustaf Rydholm
2021-05-04Nyströmer implemented but not testedGustaf Rydholm
2021-05-02Attention layer finishedGustaf Rydholm
2021-04-25Efficient net and non working transformer model.Gustaf Rydholm
2021-04-22Fixed training script, able to train vqvaeGustaf Rydholm
2021-04-15Created mappingsGustaf Rydholm
2021-04-11Working on mappingGustaf Rydholm
2021-04-08Bug fixes word piecesGustaf Rydholm
2021-04-07Fixed typing and typos, train script load config, reformattedGustaf Rydholm
2021-04-05Fix typingGustaf Rydholm
2021-04-05Pre-commit fixes, optimizer loading fixGustaf Rydholm
2021-04-05Add OmegaConf for configsGustaf Rydholm
2021-04-05Add new training loop with PyTorch Lightning, remove stale filesGustaf Rydholm
2021-04-04black reformattingGustaf Rydholm
2021-04-04Add transformer lit modelGustaf Rydholm
2021-04-04Reformat test for CERGustaf Rydholm
2021-04-04Add CER metricGustaf Rydholm
2021-03-31Removing legacy codeGustaf Rydholm
2021-03-20Inital commit for refactoring to lightningGustaf Rydholm