From 5938dd46e564b6c81e5b27680a66a93987eb11ed Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Tue, 26 Jul 2022 15:40:23 +0800 Subject: [PATCH] Fix get_transducer_model() for aishell. PR #495 introduces an error. This commit fixes it. --- egs/aishell/ASR/pruned_transducer_stateless3/decode.py | 1 + egs/aishell/ASR/pruned_transducer_stateless3/export.py | 1 + egs/aishell/ASR/pruned_transducer_stateless3/pretrained.py | 1 + 3 files changed, 3 insertions(+) diff --git a/egs/aishell/ASR/pruned_transducer_stateless3/decode.py b/egs/aishell/ASR/pruned_transducer_stateless3/decode.py index f686174f38..6aea306c8f 100755 --- a/egs/aishell/ASR/pruned_transducer_stateless3/decode.py +++ b/egs/aishell/ASR/pruned_transducer_stateless3/decode.py @@ -464,6 +464,7 @@ def main(): params = get_params() params.update(vars(args)) + params.datatang_prob = 0 assert params.decoding_method in ( "greedy_search", diff --git a/egs/aishell/ASR/pruned_transducer_stateless3/export.py b/egs/aishell/ASR/pruned_transducer_stateless3/export.py index 307895a765..566902a85e 100755 --- a/egs/aishell/ASR/pruned_transducer_stateless3/export.py +++ b/egs/aishell/ASR/pruned_transducer_stateless3/export.py @@ -157,6 +157,7 @@ def main(): params.blank_id = 0 params.vocab_size = max(lexicon.tokens) + 1 + params.datatang_prob = 0 logging.info(params) diff --git a/egs/aishell/ASR/pruned_transducer_stateless3/pretrained.py b/egs/aishell/ASR/pruned_transducer_stateless3/pretrained.py index 5cda411bc1..04a0a882a8 100755 --- a/egs/aishell/ASR/pruned_transducer_stateless3/pretrained.py +++ b/egs/aishell/ASR/pruned_transducer_stateless3/pretrained.py @@ -223,6 +223,7 @@ def main(): params.blank_id = 0 params.vocab_size = max(lexicon.tokens) + 1 + params.datatang_prob = 0 logging.info(params)