from local

This commit is contained in:
dohe0342 2023-02-02 19:11:06 +09:00
parent 64b4e778c9
commit 0ebc59866a
2 changed files with 6 additions and 1 deletions

View File

@ -470,7 +470,7 @@ def decode_dataset(
token_dict=token_dict,
)
'''
decode_one_batch_greedy(
hyps = decode_one_batch_greedy(
params=params,
model=model,
HLG=HLG,
@ -482,6 +482,11 @@ def decode_dataset(
token_dict=token_dict,
)
for i, hyp in enumerate(hyps):
print('hyp = ', hyp)
print('ref = ', texts[i])
print('')
'''
for lm_scale, hyps in hyps_dict.items():
this_batch = []