fix decode.py

This commit is contained in:
yaozengwei 2022-05-21 20:53:40 +08:00
parent 3806d2db19
commit fb8e926521

View File

@ -286,6 +286,7 @@ def decode_one_batch(
hyp_tokens = greedy_search_batch(
model=model,
encoder_out=encoder_out,
encoder_out_lens=encoder_out_lens,
)
for hyp in sp.decode(hyp_tokens):
hyps.append(hyp.split())