from local

This commit is contained in:
dohe0342 2023-02-02 19:07:31 +09:00
parent 8f19e1301e
commit 9c07cd95db
2 changed files with 3 additions and 1 deletions

View File

@ -447,7 +447,8 @@ def decode_dataset(
eos_id=eos_id,
token_dict=token_dict,
)
'''
for lm_scale, hyps in hyps_dict.items():
this_batch = []
assert len(hyps) == len(texts)
@ -463,6 +464,7 @@ def decode_dataset(
batch_str = f"{batch_idx}/{num_batches}"
logging.info(f"batch {batch_str}, cuts processed until now is {num_cuts}")
'''
return results