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)