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, encoder_out_lens=encoder_out_lens,
) )
for hyp in sp.decode(hyp_tokens): for hyp in sp.decode(hyp_tokens):
print(hyp)
hyps.append(hyp.split()) hyps.append(hyp.split())
elif params.decoding_method == "modified_beam_search": elif params.decoding_method == "modified_beam_search":
hyp_tokens = modified_beam_search( hyp_tokens = modified_beam_search(