Fix get_transducer_model() for aishell. (#497)

PR #495 introduces an error. This commit fixes it.
k2-fsa · Jul 26, 2022 · 385645d · 385645d
1 parent d3fc4b0
commit 385645d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git 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
@@ -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
@@ -223,6 +223,7 @@ def main():
 
     params.blank_id = 0
     params.vocab_size = max(lexicon.tokens) + 1
+    params.datatang_prob = 0
 
     logging.info(params)