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