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

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