from local

This commit is contained in:
dohe0342 2023-02-02 19:02:20 +09:00
parent 9991679676
commit 9a43c2a0a3
2 changed files with 1 additions and 1 deletions

View File

@ -270,7 +270,7 @@ def decode_one_batch(
topk_indexes = topk_indexes[topk_indexes != 0]
hyp = ''
for idx in topk_indexes:
hyp += token_dict[idx]
hyp += token_dict[idx.item()]
print(hyp)
exit()