summaryrefslogtreecommitdiff
path: root/text_recognizer/models/base.py
AgeCommit message (Collapse)Author
2021-10-10Update base modelGustaf Rydholm
2021-09-19Linting of Base modelGustaf 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-08Add training of VQGANGustaf Rydholm
2021-08-08Add VQGAN and loss functionGustaf Rydholm
2021-08-06Remove lr args from model, add Cosine lr, fix to vqvae stackGustaf 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-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-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-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