From ef329c07b5987a5b8024b9900402399492c68f35 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 25 Feb 2023 15:38:05 +0900
Subject: [PATCH] from local

---
 .../ASR/conformer_ctc2/.decode.py.swp         | Bin 61440 -> 65536 bytes
 egs/librispeech/ASR/conformer_ctc2/decode.py  |  13 +++++++++++++
 2 files changed, 13 insertions(+)

diff --git a/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp b/egs/librispeech/ASR/conformer_ctc2/.decode.py.swp
index d9393c9317e9b75f5cdd338bbaa20077795146fc..10145cb05b3d8f823fbebc1e75b23c4d51d7b33e 100644
GIT binary patch
delta 553
zcmXxgODKd<6u|NG&3xX>d{Z7_Gl;S9nDQu!veLwqOd(2MgB?cUV==Q~;j!_hBnumn
zN8(#4EF@AEQeuHvAW_PS75@pRe%*8LJ?C~ip?hm6q$YQ`8$4cTS%F&!O=NlXt8=bu
zJNeD7*bJ+-sni(TT8hISz9#a17g;iQHJ5$Isn#W2s3K{RoYP%nIQqYXA5kKwn7}wZ
z$U-JAsa!x4N>PGv_ilv97N#(YY^39a*Eg_^Np!=547^xH_F<w0%_y}>kz|vcTSShr
zjx|hU3~i`@6KP1ovtk?$<d}|tMK!MV+-T=tuqiZ7bemZ`QmGh2x;L>W;P-dw`GfxU
zKxc1exr7UiE8YE9bv7Jq?-KFGh@{5;j^ux0-;D4Ii)e)lPd1SwtRM&<+)(kFAo75H
zY@iWde8r30;TC(C#~6G_kC$qBj}v*s45rbGMp*F4r*3hB18iazeW*t*s*xINDylEW
I)B?{DKb+KMWB>pF

delta 319
zcmZo@U}<>3ES6*t=IN_osb|0d1Of~Uy~}?lzjW9r_J*I4cQUiUTE?u&8wJESzYvh%
zpLjuGlLAWsuMbEi6GVB|WWj>}_21YT7|sFlWFQU&VmTnb%f`U48;IutaTySM0kJv|
z|7K-icm%{ZfOrcKF9u>iAa()b4=fA}SAqBn5N`(JejqLaVlN;T0^(=P3=DUGxCMxt
zfjAzB{g@ey8GZtt^B#z=0P$rY-U`G^fVdNg(}36$h@FAh4v0ZcXJwjvs8CJAP|rX|
w!BEdoM*#$Ka^s7W6LV7Ib25ud6l{@%HcS3`&N}(tAL-4eWeJR%iz@rd0K2?Kp#T5?

diff --git a/egs/librispeech/ASR/conformer_ctc2/decode.py b/egs/librispeech/ASR/conformer_ctc2/decode.py
index 1851c057a..e3ca4ad66 100755
--- a/egs/librispeech/ASR/conformer_ctc2/decode.py
+++ b/egs/librispeech/ASR/conformer_ctc2/decode.py
@@ -411,6 +411,19 @@ def decode_one_batch(
         key = "ctc-greedy-search"
 
         return {key: hyps}
+    
+    if params.method == "greedy-search":
+        hyps = greedy_search(nnet_output, memory_key_padding_mask)
+
+        # hyps is a list of str, e.g., ['xxx yyy zzz', ...]
+        hyps = bpe_model.decode(hyps)
+
+        # hyps is a list of list of str, e.g., [['xxx', 'yyy', 'zzz'], ... ]
+        unk = bpe_model.decode(bpe_model.unk_id()).strip()
+        hyps = [[w for w in s.split() if w != unk] for s in hyps]
+        key = "ctc-greedy-search"
+
+        return {key: hyps}
 
     if params.method == "nbest-oracle":
         # Note: You can also pass rescored lattices to it.