mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 10:16:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update pretrained.py

Browse Source
This commit is contained in:
Mingshuang Luo 2021-10-14 19:00:25 +08:00
parent 40db9985ec
commit a22b638820
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
egs/librispeech/ASR/conformer_ctc/pretrained.py
View File

@ -20,7 +20,7 @@
import argparse
import logging
import math
import os
from pathlib import Path
from typing import List
import k2
@ -298,8 +298,8 @@ def main():
if params.method == "ctc-decoding":
logging.info("Use CTC decoding")
if not os.path.exists(params.bpe_model):
raise ValueError("The path to bpe.model is required!")
if not Path(params.bpe_model).exists():
raise ValueError(f"The path to {params.bpe_model} doesn't exist!")
bpe_model = spm.SentencePieceProcessor()
bpe_model.load(params.bpe_model)
@ -333,10 +333,10 @@ def main():
"whole-lattice-rescoring",
"attention-decoder",
]:
if not os.path.exists(params.HLG):
raise ValueError("The path to HLG.pt is required!")
if not os.path.exists(params.words_file):
raise ValueError("The path to words.txt is required!")
if not Path(params.HLG).exists():
raise ValueError(f"The path to {params.HLG} doesn't exist!")
if not Path(params.words_file).exists():
raise ValueError(f"The path to {params.words_file} doesn't exist!")
logging.info(f"Loading HLG from {params.HLG}")
HLG = k2.Fsa.from_dict(torch.load(params.HLG, map_location="cpu"))