mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
052a9ed91a
commit
5b6167c80d
Binary file not shown.
@ -268,7 +268,9 @@ def decode_one_batch(
|
||||
topk_log_probs, topk_indexes = nnet_output[i].topk(1)
|
||||
topk_indexes = topk_indexes.squeeze().unique_consecutive()
|
||||
topk_indexes = topk_indexes[topk_indexes != 0]
|
||||
print(topk_indexes)
|
||||
hyp = ''
|
||||
for idx in topk_indexes:
|
||||
hyp += token_dict[idx]
|
||||
|
||||
exit()
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user