From 64b4e778c9946774f5ea0f95b9c5faa4619b1268 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Thu, 2 Feb 2023 19:10:22 +0900
Subject: [PATCH] from local

---
 .../ASR/transformer_ctc/.decode.py.swp        | Bin 53248 -> 53248 bytes
 egs/aishell/ASR/transformer_ctc/decode.py     |  23 ++++++++++--------
 2 files changed, 13 insertions(+), 10 deletions(-)

diff --git a/egs/aishell/ASR/transformer_ctc/.decode.py.swp b/egs/aishell/ASR/transformer_ctc/.decode.py.swp
index 6f6d7c4f6d639f5a7f5bad5fd57d20f396abd9b4..ed89711d8576dceab9281eaedb9f3b2ac769d92b 100644
GIT binary patch
delta 314
zcmZozz}&EaSuDvQ%+puFQqO<^2m}}ywsqf5?q9P}OiF;Udb6Rx<$PXmCI*IJ7Km*1
zX2FJT#`=S73=ErrcoPu!0&y-7>jUu{RtAQxK)eKqn}IkIh#i2~9*FgUSPzJGfmj@f
z{{W4A2E?a<coPsW1mbQWt^(prAO=}kP+Fp+keiyDUsMUB<FivMH~-!Kka6;#X06G5
zExMC=S{|qsrIwTy<tb!T78EluC_n*-r<YhzkeZjG0c2}3OlD~{ntU@$krT*MuvJi3
zXPBJRYCpN8ReJKNR*lKuTCFCVv?)xEXj2ph8v-%_s4hAd%$vL}SA6rGHWg0*5I0jT

delta 185
zcmZozz}&EaSuDvQ%+puFQqO<^2m}}yCUoCUKDBzIn3Mox#b!f+%lW(lObiS>ED+g>
z&4LZxj8R9~7#J#nH~@%sfcO<F1H)n<&H&<IAl3t71|WU`)b<>Rp8@d&Al?eZ^MSYp
zh_it>3W!~R*cgbFfS8A6^NsCnOq2gKYfZLm5uI$-wUSwbfnjr7)?>!Wn{$OW-)U3v
F1OO2+DK`KB

diff --git a/egs/aishell/ASR/transformer_ctc/decode.py b/egs/aishell/ASR/transformer_ctc/decode.py
index 892888a84..0d71c3506 100755
--- a/egs/aishell/ASR/transformer_ctc/decode.py
+++ b/egs/aishell/ASR/transformer_ctc/decode.py
@@ -205,17 +205,20 @@ def decode_one_batch_greedy(
     else:
         assert HLG is None
         decoding_graph = H
+    
+    hyps = []
 
-    if params.method == 'greedy-search' or params.method == 'ctc-decoding':
-        batch_size = nnet_output.size(0)
-        for i in range(batch_size):
-            topk_log_probs, topk_indexes = nnet_output[i].topk(1)
-            topk_indexes = topk_indexes.squeeze().unique_consecutive()
-            topk_indexes = topk_indexes[topk_indexes != 0]
-            hyp = ''
-            for idx in topk_indexes:
-                hyp += token_dict[idx.item()]
-            print(hyp) 
+    batch_size = nnet_output.size(0)
+    for i in range(batch_size):
+        topk_log_probs, topk_indexes = nnet_output[i].topk(1)
+        topk_indexes = topk_indexes.squeeze().unique_consecutive()
+        topk_indexes = topk_indexes[topk_indexes != 0]
+        hyp = ''
+        for idx in topk_indexes:
+            hyp += token_dict[idx.item()]
+        hyps.append(hyp)
+
+    return hyps
 
 
 def decode_one_batch(