from local

This commit is contained in:
dohe0342 2023-02-02 18:55:50 +09:00
parent 2c7c3cc221
commit 9a62cd8fb7
2 changed files with 1 additions and 1 deletions

View File

@ -267,7 +267,7 @@ 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.size())
print(topk_indexes)
#print(topk_indexes.squeeze())
exit()