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

Minor fixes.

Browse Source
This commit is contained in:
Fangjun Kuang 2022-01-07 23:04:18 +08:00
parent ec9927c4bf
commit 5ff660c84d
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
egs/librispeech/ASR/local/compile_lg.py
View File

@ -17,7 +17,7 @@
"""
This script takes as input lang_dir and generates HLG from
This script takes as input lang_dir and generates LG from
- L, the lexicon, built from lang_dir/L_disambig.pt
@ -66,7 +66,7 @@ def compile_LG(lang_dir: str) -> kaldifst.StdVectorFst:
assert "#0" in tokens
assert "#0" in words
token_disambig_id = tokens.find("#02")
token_disambig_id = tokens.find("#0")
word_disambig_id = words.find("#0")
L = k2.Fsa.from_dict(torch.load(f"{lang_dir}/L_disambig.pt"))
@ -84,7 +84,7 @@ def compile_LG(lang_dir: str) -> kaldifst.StdVectorFst:
)
kaldifst.arcsort(G, sort_type="ilabel")
logging.info(f"Composing LG")
logging.info("Composing LG")
LG = kaldifst.compose(
L,
G,
@ -93,10 +93,10 @@ def compile_LG(lang_dir: str) -> kaldifst.StdVectorFst:
connect=True,
)
logging.info(f"Determinize star LG")
logging.info("Determinize star LG")
kaldifst.determinize_star(LG)
logging.info(f"minimizeencoded")
logging.info("minimizeencoded")
kaldifst.minimize_encoded(LG)
# Set all disambig IDs to eps