summaryrefslogtreecommitdiff
path: root/text_recognizer
AgeCommit message (Collapse)Author
2021-09-30Bug fix for loading pretrained vq encoderGustaf Rydholm
2021-09-30Set leaky relu inplace to falseGustaf Rydholm
2021-09-30Rename context to trg in transformerGustaf Rydholm
2021-09-30Rename vqloss to commitment loss in vqvae networkGustaf Rydholm
2021-09-30Refactor residual blockGustaf Rydholm
2021-09-30Add num res blocks as a variableGustaf Rydholm
2021-09-30Rename to commitment lossGustaf Rydholm
2021-09-30Bug fix transformer dim, comment CER/WER metricsGustaf Rydholm
2021-09-30Lint sentence_generatorGustaf Rydholm
2021-09-30Lint emnist mapping importsGustaf Rydholm
2021-09-30Rename vqloss to commitment lossGustaf Rydholm
2021-09-30Update GAN loss with sigmoid fnGustaf Rydholm
2021-09-19Linting of Base modelGustaf Rydholm
2021-09-19Add loading of encoder in vq transformer networkGustaf Rydholm
2021-09-19Add ability to set use norm in vqvaeGustaf Rydholm
2021-09-19Add weight for commitment loss in vq transformerGustaf Rydholm
2021-09-19Linting of iam paragraphs and preprocessorGustaf Rydholm
2021-09-19Linting of eminst and iam linesGustaf Rydholm
2021-09-19Linting of vqgan lossGustaf Rydholm
2021-09-19Remove normalization in a hope to fix instabilitiesGustaf Rydholm
2021-09-19Add more detailed description of label smoothingGustaf Rydholm
2021-09-18Add comment for quantizationGustaf Rydholm
2021-09-18Remove TODO in conv transformerGustaf Rydholm
2021-09-18Add Vq transfomer modelGustaf Rydholm
2021-09-18Add more residual blocks in vqvaeGustaf Rydholm
2021-09-18Add resize attribute to IAM paragraphsGustaf Rydholm
2021-09-11Disable group norm in vqvae en/decoderGustaf 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-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-08-03Fix bugs in converting text in mappings, add missing word_piece arg in ↵Gustaf Rydholm
datamodule
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-08-02Fixed wandb callback bug, add word_pieces arg to iam paragraphGustaf 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-28Reformatting with attrs, config for encoder and decoderGustaf Rydholm
2021-07-23Remove dinoGustaf Rydholm