Fix get_transducer_model() for aishell. (#497)

PR #495 introduces an error. This commit fixes it.
This commit is contained in:
Fangjun Kuang 2022-07-26 15:42:21 +08:00 committed by GitHub
parent d3fc4b031e
commit 385645d533
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 0 deletions

View File

@ -464,6 +464,7 @@ def main():
params = get_params() params = get_params()
params.update(vars(args)) params.update(vars(args))
params.datatang_prob = 0
assert params.decoding_method in ( assert params.decoding_method in (
"greedy_search", "greedy_search",

View File

@ -157,6 +157,7 @@ def main():
params.blank_id = 0 params.blank_id = 0
params.vocab_size = max(lexicon.tokens) + 1 params.vocab_size = max(lexicon.tokens) + 1
params.datatang_prob = 0
logging.info(params) logging.info(params)

View File

@ -223,6 +223,7 @@ def main():
params.blank_id = 0 params.blank_id = 0
params.vocab_size = max(lexicon.tokens) + 1 params.vocab_size = max(lexicon.tokens) + 1
params.datatang_prob = 0
logging.info(params) logging.info(params)