summaryrefslogtreecommitdiff
path: root/text_recognizer/models
AgeCommit message (Collapse)Author
2022-06-18Refator loading of scheduler and optimGustaf Rydholm
2022-06-15Calculate cer and acc in val stepGustaf Rydholm
2022-06-12Remove conformerGustaf Rydholm
2022-06-11Fix emnist mappingGustaf Rydholm
2022-06-10Add importsGustaf Rydholm
2022-06-09Remove abstract mappingGustaf Rydholm
2022-06-08Fix bug in lit conformerGustaf Rydholm
2022-06-06Add lit conformer modelGustaf Rydholm
2022-06-06Rename lit modelsGustaf Rydholm
2022-06-05Remove attrsGustaf Rydholm
2022-06-01Replace attr with attrsGustaf Rydholm
2022-02-03chore: remove unused code block in lit trans modelGustaf Rydholm
2021-11-21Format filesGustaf Rydholm
2021-11-21Remove VQVAE stuff, did not work...Gustaf Rydholm
2021-11-17Update vq transformer lit modelGustaf Rydholm
2021-10-27Fix import of criterion in VQGANGustaf Rydholm
2021-10-27Fix mapping import in base modelGustaf Rydholm
2021-10-27Remove Barlow TwinsGustaf Rydholm
2021-10-27Fix imports in transformer modelGustaf Rydholm
2021-10-24Format of filesGustaf Rydholm
2021-10-11Remove mapping from transformer lit modelsGustaf Rydholm
2021-10-11Add mapping to base lit modelGustaf Rydholm
2021-10-11Refactor barlow twins lit modelGustaf Rydholm
2021-10-10Update transformer modelsGustaf Rydholm
2021-10-10Update base modelGustaf Rydholm
2021-10-10Update barlow modelGustaf Rydholm
2021-10-10Refactor of vq modelsGustaf Rydholm
2021-10-07Add adaptive pool between encoder and projectorGustaf Rydholm
2021-10-07Add Barlow Twins network and training proceduerGustaf Rydholm
2021-10-07Remove unused importGustaf Rydholm
2021-09-30Rename to commitment lossGustaf Rydholm
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