from local

This commit is contained in:
dohe0342 2023-02-02 18:56:14 +09:00
parent 9a62cd8fb7
commit f5678f57ef
2 changed files with 0 additions and 1 deletions

View File

@ -268,7 +268,6 @@ def decode_one_batch(
topk_indexes = topk_indexes.squeeze().unique_consecutive() topk_indexes = topk_indexes.squeeze().unique_consecutive()
topk_indexes = topk_indexes[topk_indexes != 0] topk_indexes = topk_indexes[topk_indexes != 0]
print(topk_indexes) print(topk_indexes)
#print(topk_indexes.squeeze())
exit() exit()