From 3b48f52b710654ccd66d0234f29a9dc0619462ef Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 14 Feb 2023 18:40:32 +0900
Subject: [PATCH] from local

---
 egs/tedlium2/ASR/conformer_ctc3/.train.py.swp | Bin 61440 -> 61440 bytes
 .../ASR/conformer_ctc3/.transformer.py.swp    | Bin 61440 -> 61440 bytes
 .../ASR/conformer_ctc3/transformer.py         |   3 +--
 3 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp
index 075c51b537f37163b7b9ce67f09651e7f518e998..02369516aac060c0c58cd1ba12fcef7ac93f85d0 100644
GIT binary patch
delta 32
mcmZp8z})bFSv1KY%+puFQqO<^2m}}y%EMkKzuqYN`aJ-jR0_uc

delta 32
mcmZp8z})bFSv1KY%+puFQqO<^2m}}y3c_9|AJ{1R`aJ-iJ_?lp

diff --git a/egs/tedlium2/ASR/conformer_ctc3/.transformer.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.transformer.py.swp
index 74a458f1df7a40ab8683cbf7e1c7eae2477170f4..50f0cb9c7b0bbbfa9d299853d4697df34f4ad8d8 100644
GIT binary patch
delta 181
zcmZp8z})bFd4qufW7uRv0eMD|&4B`&^?9wB7#P+uGcYg+FffE|7IX+^<TYktVE6_U
z0tqkOT<B=RG`Ue<baJcn>d776e4C41CUWy$=3!v4=Y?n!nJgF}>{P_hz+eKz#y~6w
z!~#J4osWUx5D+f~;(Q=>2VyrMwg6&nApQn4;T{lg0b+aJ%?kx@@@&qJwPu=Z>SC~Y
IReab40A*Jv%>V!Z

delta 179
zcmZp8z})bFd4qufW5i@b0eME@&4B`&^?Cm?GB8v!GcYg+Ffc@H7IX+^<TYerV9)^K
ze?Z}bn+qLHm?qzH?q+0|To!LKxx<@pbFs@rZeDg?28Mhf25A(YEEpi{l)}%zkPO6r
zK<og->OlO7kAdMT5YGqVav&}P;tU{;24XcJ<^$p<P?OqtH!l?A<lSr^Yt6KIReab4
E02;+5vH$=8

diff --git a/egs/tedlium2/ASR/conformer_ctc3/transformer.py b/egs/tedlium2/ASR/conformer_ctc3/transformer.py
index 4ff9062fe..6743734a6 100644
--- a/egs/tedlium2/ASR/conformer_ctc3/transformer.py
+++ b/egs/tedlium2/ASR/conformer_ctc3/transformer.py
@@ -186,14 +186,13 @@ class Transformer(nn.Module):
         encoder_memory, memory_key_padding_mask = self.run_encoder(
             x, supervision, warmup
         )
-        x = self.ctc_output(encoder_memory)
-        
         if type(encoder_memory) == tuple:
             (encoder_memory, layer_outputs) = encoder_memory
             layer_outputs = [self.ctc_output(x) for x in layer_outputs]
 
             return (x, layer_outputs), encoder_memory, memory_key_padding_mask
         else:
+            x = self.ctc_output(encoder_memory)
             return x, encoder_memory, memory_key_padding_mask
 
     def run_encoder(