mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

add modified-beam-search for streaming decoding

Browse Source
This commit is contained in:
luomingshuang 2022-07-25 16:11:59 +08:00
parent fee72b9cbb
commit 4bf839227c
1 changed files with 0 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
egs/wenetspeech/ASR/pruned_transducer_stateless5/streaming_decode.py
View File

@ -391,21 +391,13 @@ def decode_dataset(
decode_stream.ground_truth = cut.supervisions[0].text
decode_streams.append(decode_stream)
print(len(decode_streams), params.num_decode_streams)
while len(decode_streams) >= params.num_decode_streams:
finished_streams = decode_one_chunk(
params=params, model=model, decode_streams=decode_streams
)
print("abcdefg...")
for i in sorted(finished_streams, reverse=True):
hyp = decode_streams[i].decoding_result()
print("ref: ", list(decode_streams[i].ground_truth))
print(
"hyp: ",
[lexicon.token_table[idx] for idx in hyp],
[idx for idx in hyp],
)
decode_results.append(
(
list(decode_streams[i].ground_truth),
@ -424,8 +416,6 @@ def decode_dataset(
)
for i in sorted(finished_streams, reverse=True):
hyp = decode_streams[i].decoding_result()
print("ref: ", list(decode_streams[i].ground_truth))
print("hyp: ", [lexicon.token_table[idx] for idx in hyp])
decode_results.append(
(
list(decode_streams[i].ground_truth),