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, token_dict=token_dict,
) )
''' '''
decode_one_batch_greedy( hyps = decode_one_batch_greedy(
params=params, params=params,
model=model, model=model,
HLG=HLG, HLG=HLG,
@ -481,6 +481,11 @@ def decode_dataset(
eos_id=eos_id, eos_id=eos_id,
token_dict=token_dict, 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(): for lm_scale, hyps in hyps_dict.items():