mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-02-02 19:07:21 +09:00
parent d936ec48a9
commit 8f19e1301e
2 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/aishell/ASR/transformer_ctc/.decode.py.swp
View File

Binary file not shown.

14
egs/aishell/ASR/transformer_ctc/decode.py
View File

@ -423,7 +423,7 @@ def decode_dataset(
logging.info(f"decoding {batch_idx} th batch") logging.info(f"decoding {batch_idx} th batch")
texts = batch["supervisions"]["text"] texts = batch["supervisions"]["text"]
cut_ids = [cut.id for cut in batch["supervisions"]["cut"]] cut_ids = [cut.id for cut in batch["supervisions"]["cut"]]
'''
hyps_dict = decode_one_batch( hyps_dict = decode_one_batch(
params=params, params=params,
model=model, model=model,
@ -435,6 +435,18 @@ def decode_dataset(
eos_id=eos_id, eos_id=eos_id,
token_dict=token_dict, token_dict=token_dict,
) )
'''
decode_one_batch_greedy(
params=params,
model=model,
HLG=HLG,
H=H,
batch=batch,
lexicon=lexicon,
sos_id=sos_id,
eos_id=eos_id,
token_dict=token_dict,
)
for lm_scale, hyps in hyps_dict.items(): for lm_scale, hyps in hyps_dict.items():
this_batch = [] this_batch = []