From 41c3e99fe57874ba1855c893bf47087d474ec6b8 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 25 Oct 2021 22:32:10 +0200 Subject: Updates to configs --- training/conf/criterion/transducer.yaml | 14 ++++++++++++++ training/conf/criterion/vqgan_loss.yaml | 1 - 2 files changed, 14 insertions(+), 1 deletion(-) create mode 100644 training/conf/criterion/transducer.yaml (limited to 'training/conf/criterion') 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: [ "", "", "

", "\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 - -- cgit v1.2.3-70-g09d2