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