Update decode.py

This commit is contained in:
Mingshuang Luo 2021-10-28 15:36:31 +08:00
parent 5e7c733b11
commit e2bb9b4350

View File

@ -409,7 +409,7 @@ def main():
if params.method in ["nbest-rescoring", "whole-lattice-rescoring"]: if params.method in ["nbest-rescoring", "whole-lattice-rescoring"]:
if not (params.lm_dir / "G_4_gram.pt").is_file(): if not (params.lm_dir / "G_4_gram.pt").is_file():
logging.info("Loading G_4_gram.fst.txt") logging.info("Loading G_4_gram.fst.txt")
logging.warning("It may take 8 minutes.") logging.warning("It may take 20 seconds.")
with open(params.lm_dir / "G_4_gram.fst.txt") as f: with open(params.lm_dir / "G_4_gram.fst.txt") as f:
first_word_disambig_id = lexicon.word_table["#0"] first_word_disambig_id = lexicon.word_table["#0"]