From 00207b7f779ff336f683a5482dd041d38d526342 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 14 Feb 2023 14:58:54 +0900
Subject: [PATCH] from local

---
 egs/tedlium2/ASR/conformer_ctc3/.train.py.swp | Bin 57344 -> 57344 bytes
 .../ASR/conformer_ctc3/.transformer.py.swp    | Bin 57344 -> 57344 bytes
 .../ASR/conformer_ctc3/transformer.py         |   7 +++++--
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.train.py.swp
index bdf9ff5ee75c4616cc12e471d7b4cc45240d09a1..bfa70faa6dc2b05ceb863b735590116bcbe5fca8 100644
GIT binary patch
delta 33
ncmZoTz}#?vSuDvQ%+puFQqO<^2m}}ya+O{u+aB5|_Ub+Won;D6

delta 33
ncmZoTz}#?vSuDvQ%+puFQqO<^2m}}yw3S{b-#xfd?A3h$o{S2s

diff --git a/egs/tedlium2/ASR/conformer_ctc3/.transformer.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.transformer.py.swp
index feb52c4b1ae5bf3ab454862eb9e7e2f2e3a0d33a..2fb51e3bf975a9f976a92ea644449be806ad6998 100644
GIT binary patch
delta 229
zcmZoTz}#?vd4qufW5i@b0eME@&4B`&^?Cm?GB8v!GcYg+Ffc@H7IX+^6f|UEV9)^K
ze?Z}bKwP=G)bTOn<aN%9K)ikOrFh-R4c=m#i(MviC$aM~FysR<NV_nQ0)hYh3=A89
zco7h%0C6%9`vI{75UT_6Cq4#-t3W&-h|7Vv42Uy;I2wr6fS3=6pFquR<K5gS$jLkT
rtRm0m`LQ1vc@==5D7B=tC{Lk6XYxcBjmf)vr6*5_ci#Lzp0NP{Kz%jU

delta 195
zcmZoTz}#?vd4qufW7uRv0eMED&4B`&^?9wB7#P+uGcYg+FffE|7IX+^6f|aGVE6_U
z0tr6{;<cMg9Un7J=8sRD+~T}ua)&qH=3<wL+)?_x3=ETj7^GDQNP)nAeg=liK->Yu
z?Lb@v#3?}R4a8zV{F9G?;V=--2I5&j+ylh*K<ov?MnKE~#FKe9Hwv2aZqASW%Q$&Y
Pyz^v37oE-j;~5(OwYn>N

diff --git a/egs/tedlium2/ASR/conformer_ctc3/transformer.py b/egs/tedlium2/ASR/conformer_ctc3/transformer.py
index afc46e311..ed1bb192e 100644
--- a/egs/tedlium2/ASR/conformer_ctc3/transformer.py
+++ b/egs/tedlium2/ASR/conformer_ctc3/transformer.py
@@ -186,12 +186,15 @@ 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]
 
-        x = self.ctc_output(encoder_memory)
-        return (x, layer_outputs), encoder_memory, memory_key_padding_mask
+            return (x, layer_outputs), encoder_memory, memory_key_padding_mask
+        else:
+            return x, encoder_memory, memory_key_padding_mask
 
     def run_encoder(
         self,