From 2c7c3cc221e091027d43ae1a510e8063946f161a Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Thu, 2 Feb 2023 18:55:20 +0900
Subject: [PATCH] from local

---
 .../ASR/transformer_ctc/.decode.py.swp        | Bin 36864 -> 36864 bytes
 egs/aishell/ASR/transformer_ctc/decode.py     |   5 ++++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/egs/aishell/ASR/transformer_ctc/.decode.py.swp b/egs/aishell/ASR/transformer_ctc/.decode.py.swp
index 390afaedb404632e24936b519b2438cab324f470..d9f99841fe8f8886df8da2a40c1118af4ab2359c 100644
GIT binary patch
delta 227
zcmZozz|^pSNi4}A%+puFQqO<^2m}}yBsy*<doJB5W+lL=w>eN?vp??^P6mb!E{Lq&
zWWj{Rl1|(V4D3Ms04TZ@h*tpdR3Pryd^TY>=VYU7OGf3%+1Xk^EE1bpm8zks$)JD)
zO7aV`<1_P8QY%u66>JrdxzWfR1w~s0gIIi8Cl_XmiRhK)WfqpE#wX|J6{jYbmSmQt
LYHV)Lu2ca4H&a01

delta 113
zcmZozz|^pSNi4}A%+puFQqO<^2m}}y{<Pmt{;_1En3Vvd_U1r=&HlVuTnr3G+z?sq
z$$|-sCF6m-ARx8}Vvx)$po#}Tyk+y*gx#EzQ{z@JDJV?t$g-ULCQD@UwJfpCQ?s*G
E0CbTbvj6}9

diff --git a/egs/aishell/ASR/transformer_ctc/decode.py b/egs/aishell/ASR/transformer_ctc/decode.py
index 4ce5774d3..f63a1411b 100755
--- a/egs/aishell/ASR/transformer_ctc/decode.py
+++ b/egs/aishell/ASR/transformer_ctc/decode.py
@@ -265,7 +265,10 @@ 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.squeeze())
+            topk_indexes = topk_indexes.squeeze().unique_consecutive()
+            topk_indexes = topk_indexes[topk_indexes != 0]
+            print(topk_indexes.size())
+            #print(topk_indexes.squeeze())
             
             exit()