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

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

diff --git a/egs/aishell/ASR/transformer_ctc/.decode.py.swp b/egs/aishell/ASR/transformer_ctc/.decode.py.swp
index 35dde24b68eafb597a966f254f286dcc560fd8f2..e11d14c1cb50ceb94287744aa7b9506b84a4b4c8 100644
GIT binary patch
delta 320
zcmXBO!7BuD7{>8uX4arae$<{EOkDP$ZKV_^Noo&DS;R#dwA*C&vT2=+tdOHWV7Hu<
zi+*WuZo6mYvdCdC$lu`LTi*ISJ@s7Pr0FG1FTI=F$ghNR*_=2dMc41q*+Y8FI5j(M
zUMfSzm4ej&ShFL^uygV;D7A5i2F_8(0rJSegNxSzsf$aLP{aae5unpAo#F^F)=|JB
zCgJ1PmwM=+?Q2uVlvPaOr%$>?5o^do_~z#?eBv38XrPW5+gQR3CJ>}-x8=JgtXAUv
h^3Gl<s?LX`&0IXWH}-JU7RvXIeQRwwws$qY{0FF@JFx%&

delta 287
zcmXBNt!qMY6vpv$Uz`pU@4^M=41$Oz(O_6H4jssXD069O62=4tS@Q-##4Omn(K>#?
z$w1TofZ?#1EQVoJF!~)k@PPx*;XICOIIdyW(v?CndYeiMV~IQtH}$7?`^M6$Dr2ph
zfvHC%>HQC@>dFaf-R-5w2rtOv0!yDr4^7-70uu&w?0NBxPq^r#gJ(P-hrgNx8dSy+
z!mlYZ!5bQgnNkv)<zR^H@P!Y&qKqQ4xJD31om;Fi#~5w2P(U6hP5sQW9xbZ;y@jrJ
HgYexM5eF`D

diff --git a/egs/aishell/ASR/transformer_ctc/decode.py b/egs/aishell/ASR/transformer_ctc/decode.py
index c2cc939dd..020010419 100755
--- a/egs/aishell/ASR/transformer_ctc/decode.py
+++ b/egs/aishell/ASR/transformer_ctc/decode.py
@@ -174,6 +174,37 @@ def decode_one_batch_greedy(
     eos_id: int,
     token_dict: dict,
     ) -> Dict[str, List[List[int]]]:
+    
+    if HLG is not None:
+        device = HLG.device
+    else:
+        device = H.device
+
+    feature = batch["inputs"]
+    assert feature.ndim == 3
+    feature = feature.to(device)
+    # at entry, feature is (N, T, C)
+
+    supervisions = batch["supervisions"]
+
+    nnet_output, memory, memory_key_padding_mask = model(feature, supervisions)
+    # nnet_output is (N, T, C)
+
+    supervision_segments = torch.stack(
+        (
+            supervisions["sequence_idx"],
+            supervisions["start_frame"] // params.subsampling_factor,
+            supervisions["num_frames"] // params.subsampling_factor,
+        ),
+        1,
+    ).to(torch.int32)
+
+    if H is None:
+        assert HLG is not None
+        decoding_graph = HLG
+    else:
+        assert HLG is None
+        decoding_graph = H
 
     if params.method == 'greedy-search' or params.method == 'ctc-decoding':
         batch_size = nnet_output.size(0)