summaryrefslogtreecommitdiff
path: root/text_recognizer/networks
AgeCommit message (Collapse)Author
2021-10-28Fix bug with local attentionGustaf Rydholm
2021-10-28Refactor attention moduleGustaf Rydholm
2021-10-27Remove Barlow TwinsGustaf Rydholm
2021-10-27Add axial embeddingGustaf Rydholm
2021-10-27Remove ViTGustaf Rydholm
2021-10-27Remove default transformerGustaf Rydholm
2021-10-27Add comments to transformer modulesGustaf Rydholm
2021-10-27Clean up local attention, add comments and typesGustaf Rydholm
2021-10-27Add local attn in transformer layerGustaf Rydholm
2021-10-27Remove unused import and add comments in attn moduleGustaf Rydholm
2021-10-27Remove imports from __init__ in transformer networkGustaf Rydholm
2021-10-27Rename transformer embeddingsGustaf Rydholm
2021-10-24Add local attentionGustaf Rydholm
2021-10-24Remove unused import in conv transformerGustaf Rydholm
2021-10-24Refator attentionGustaf Rydholm
2021-10-24Format of filesGustaf Rydholm
2021-10-11Add BarlowNetworkGustaf Rydholm
2021-10-10Remove importsGustaf Rydholm
2021-10-07LintGustaf Rydholm
2021-10-07Add Barlow Twins network and training proceduerGustaf Rydholm
2021-10-03Refactor rotary embeddingGustaf Rydholm
2021-10-03Fix rotary embeddingGustaf Rydholm
2021-10-03Lint filesGustaf Rydholm
2021-09-30Major bug fix in attention layerGustaf Rydholm
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-19Add loading of encoder in vq transformer networkGustaf Rydholm
2021-09-19Add ability to set use norm in vqvaeGustaf 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-11Disable group norm in vqvae en/decoderGustaf Rydholm
2021-09-11Rename vq loss to commitment lossGustaf Rydholm
2021-08-08Add VQGAN and loss functionGustaf 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-03Training working, multiple bug fixesGustaf 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-29Configs, refactor with attrs, fix attr bug in iamGustaf Rydholm
2021-07-28Reformatting with attrs, config for encoder and decoderGustaf Rydholm
2021-07-23Remove nystromerGustaf Rydholm
2021-07-23Complete cnn-transformer network, not testedGustaf Rydholm