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

fix MMI decode graph

Browse Source
This commit is contained in:
PingFeng Luo 2021-12-01 11:22:25 +08:00
parent a54f9a9b41
commit 4b6edaa4a3
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
egs/aishell/ASR/conformer_mmi/decode.py
View File

@ -39,6 +39,7 @@ from icefall.decode import (
)
from icefall.env import get_env_info
from icefall.lexicon import Lexicon
from icefall.mmi_graph_compiler import MmiTrainingGraphCompiler
from icefall.utils import (
AttributeDict,
get_texts,
@ -493,11 +494,12 @@ def main():
logging.info(f"device: {device}")
graph_compiler = BpeCtcTrainingGraphCompiler(
params.lang_dir,
graph_compiler = MmiTrainingGraphCompiler(
args.lang_dir,
device=device,
sos_token="<sos/eos>",
eos_token="<sos/eos>",
oov="<UNK>",
sos_id=1,
eos_id=1,
)
sos_id = graph_compiler.sos_id
eos_id = graph_compiler.eos_id