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
2f8218bff5
commit
aa5b891b08
Binary file not shown.
@ -265,7 +265,8 @@ def decode_one_batch(
|
|||||||
batch_size = nnet_output.size(0)
|
batch_size = nnet_output.size(0)
|
||||||
for i in range(batch_size):
|
for i in range(batch_size):
|
||||||
topk_log_probs, topk_indexes = nnet_output[i].topk(1)
|
topk_log_probs, topk_indexes = nnet_output[i].topk(1)
|
||||||
print(topk_indexes)
|
print(topk_indexes.size())
|
||||||
|
|
||||||
exit()
|
exit()
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user