from local
This commit is contained in:
parent
9991679676
commit
9a43c2a0a3
Binary file not shown.
@ -270,7 +270,7 @@ def decode_one_batch(
|
|||||||
topk_indexes = topk_indexes[topk_indexes != 0]
|
topk_indexes = topk_indexes[topk_indexes != 0]
|
||||||
hyp = ''
|
hyp = ''
|
||||||
for idx in topk_indexes:
|
for idx in topk_indexes:
|
||||||
hyp += token_dict[idx]
|
hyp += token_dict[idx.item()]
|
||||||
print(hyp)
|
print(hyp)
|
||||||
exit()
|
exit()
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user