from local

This commit is contained in:
dohe0342 2023-06-09 17:39:33 +09:00
parent 8d946917ed
commit d51a24d081
2 changed files with 1 additions and 0 deletions

View File

@ -444,6 +444,7 @@ def decode_one_batch(
encoder_out_lens=encoder_out_lens,
)
for hyp in sp.decode(hyp_tokens):
print(hyp)
hyps.append(hyp.split())
elif params.decoding_method == "modified_beam_search":
hyp_tokens = modified_beam_search(