Update decode.py

This commit is contained in:
jinzr 2023-11-09 11:37:18 +08:00
parent fb541ec60c
commit 16499a5ef6

View File

@ -520,6 +520,7 @@ def decode_one_batch(
encoder_out=encoder_out,
encoder_out_lens=encoder_out_lens,
beam=params.beam_size,
context_graph=context_graph,
)
for hyp in sp.decode(hyp_tokens):
hyps.append(hyp.split())