from local

This commit is contained in:
dohe0342 2023-02-02 18:53:00 +09:00
parent 2f8218bff5
commit aa5b891b08
2 changed files with 2 additions and 1 deletions

View File

@ -265,7 +265,8 @@ 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)
print(topk_indexes.size())
exit()