summaryrefslogtreecommitdiff
path: root/training
AgeCommit message (Expand)Author
2021-09-11Rename vq loss to commitment lossGustaf Rydholm
2021-08-30Update dependencies, update to vqvae paramsGustaf 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-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 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 datamo...Gustaf 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-08-02Fixed wandb callback bug, add word_pieces arg to iam paragraphGustaf 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-08Move callbacks to training folder, refactorGustaf Rydholm
2021-07-06Refactoring data with attrs and refactor conf for hydraGustaf Rydholm
2021-07-05Finalizing training loop with hydraGustaf Rydholm
2021-07-05Working on new hydra implementationGustaf Rydholm
2021-07-04Remove unused codeGustaf Rydholm
2021-07-04Add wandb callbacksGustaf Rydholm
2021-06-29Refactor of config, more granularGustaf Rydholm
2021-06-27Auto lrGustaf Rydholm
2021-06-27RenamingGustaf Rydholm
2021-06-27Save experiment config to log dirlightning-refactorGustaf Rydholm
2021-06-27Changed structure of callbacksGustaf Rydholm
2021-05-02Attention layer finishedGustaf Rydholm
2021-05-02Working on attentionGustaf Rydholm
2021-05-02Implemented training script with hydraGustaf Rydholm
2021-04-25Efficient net and non working transformer model.Gustaf Rydholm
2021-04-24Implementing CoaT transformer, continue tomorrow...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-11Changed project layout againGustaf Rydholm
2021-04-08Bug fixes word piecesGustaf Rydholm
2021-04-07Typing fixed, tasks removed from noxfileGustaf Rydholm
2021-04-07Fixed typing and typos, train script load config, reformattedGustaf Rydholm
2021-04-06Refactor train scriptGustaf Rydholm
2021-04-06Fix network argsGustaf Rydholm
2021-04-05Pre-commit fixes, optimizer loading fixGustaf Rydholm
2021-04-05Completed first draft for training loop with PyTorch LightningGustaf Rydholm
2021-04-05Add new training loop with PyTorch Lightning, remove stale filesGustaf Rydholm
2021-03-20Inital commit for refactoring to lightningGustaf Rydholm