diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-25 22:32:10 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-25 22:32:10 +0200 |
| commit | 41c3e99fe57874ba1855c893bf47087d474ec6b8 (patch) | |
| tree | 2ba286209251f7ec8d7ab4cf061cffe5de86f398 /training/conf/criterion | |
| parent | da29c1cf4d062087f1b29dc9402ee6384203b690 (diff) | |
Updates to configs
Diffstat (limited to 'training/conf/criterion')
| -rw-r--r-- | training/conf/criterion/transducer.yaml | 14 | ||||
| -rw-r--r-- | training/conf/criterion/vqgan_loss.yaml | 1 |
2 files changed, 14 insertions, 1 deletions
diff --git a/training/conf/criterion/transducer.yaml b/training/conf/criterion/transducer.yaml new file mode 100644 index 0000000..7e661a7 --- /dev/null +++ b/training/conf/criterion/transducer.yaml @@ -0,0 +1,14 @@ +_target_: text_recognizer.criterions.transducer.Transducer +preprocessor: + _target_: text_recognizer.data.utils.iam_preprocessor.Preprocessor + num_features: 1000 + tokens: iamdb_1kwp_tokens_1000.txt + lexicon: iamdb_1kwp_lex_1000.txt + use_words: false + prepend_wordsep: false + special_tokens: [ "<s>", "<e>", "<p>", "\n" ] +ngram: 0 +# transitions: 1kwp_prune_0_10_optblank.bin +blank: "optional" +allow_repeats: true +reduction: "none" diff --git a/training/conf/criterion/vqgan_loss.yaml b/training/conf/criterion/vqgan_loss.yaml index f983f6f..43e05cd 100644 --- a/training/conf/criterion/vqgan_loss.yaml +++ b/training/conf/criterion/vqgan_loss.yaml @@ -9,4 +9,3 @@ discriminator: num_layers: 3 vq_loss_weight: 1.0 discriminator_weight: 1.0 - |