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

---
 .../ASR/transformer_ctc/.decode.py.swp        | Bin 49152 -> 49152 bytes
 egs/aishell/ASR/transformer_ctc/decode.py     |  24 ++++++++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/egs/aishell/ASR/transformer_ctc/.decode.py.swp b/egs/aishell/ASR/transformer_ctc/.decode.py.swp
index 612c1316e7910b86d667ac97357779c869a07b3c..eb3f1c45c01652f7514c3bf9f1ccb3d1b368f02f 100644
GIT binary patch
delta 234
zcmZo@U~Xt&7ELk;^Ym4))H7fJ0s#hwOI^2<S8f!Q7GR9rY$$L!pLYc-1A_)TL^5)+
zV8i43(_9P;8-REg5Z4278W4v9u{{v$0kJF)vjg!XP6mdDKs*(QbAZ?bh}nR6KL-Ot
z3lQf6u@?}_05J;?GXpUb5WfN1b`gko1MvbNZU^ErAWi^ce|83L20I`F<Ro?W&2^0{
lSs4{3^L0Jg+?{ogaq|8Bia^c{_W1Oo)YO#9&Cj}|OaUPRFH-;j

delta 216
zcmZo@U~Xt&7ELk;^Ym4))H7fJ0s#hwkgnUwvKvLE1sEMS8wy;`=Y7Y)z`(}^k#w9a
z*zmai5|FnYh!+BJ8xZFLaSRZ<0kJ6%s{k=bhd2;#=44=)3&eAPI0T4UfOtR9#9|;;
z2jcha3=GSFI0}fhfmjiUg@O1U8w0~#AnpNTYao8d%E0gzh##;ra5EeOGIj%T5$oo<
W#+9s_Irlzinyk_OVDq~!AyWXd{VBl!

diff --git a/egs/aishell/ASR/transformer_ctc/decode.py b/egs/aishell/ASR/transformer_ctc/decode.py
index bc0fc279f..ce76c2fec 100755
--- a/egs/aishell/ASR/transformer_ctc/decode.py
+++ b/egs/aishell/ASR/transformer_ctc/decode.py
@@ -163,6 +163,30 @@ def get_params() -> AttributeDict:
     return params
 
 
+def decode_one_batch_greedy(
+    params: AttributeDict,
+    model: nn.Module,
+    HLG: Optional[k2.Fsa],
+    H: Optional[k2.Fsa],
+    batch: dict,
+    lexicon: Lexicon,
+    sos_id: int,
+    eos_id: int,
+    token_dict: dict,
+    ) -> Dict[str, List[List[int]]]:
+
+    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) 
+
+
 def decode_one_batch(
     params: AttributeDict,
     model: nn.Module,