from local

This commit is contained in:
dohe0342 2023-06-09 17:40:55 +09:00
parent d51a24d081
commit 6d51037a54
2 changed files with 3 additions and 2 deletions

View File

@ -444,7 +444,6 @@ 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(
@ -555,7 +554,9 @@ def decode_dataset(
word_table=word_table,
batch=batch,
)
print(hyps_dict)
exit()
for name, hyps in hyps_dict.items():
this_batch = []
assert len(hyps) == len(texts)