diff options
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 - |