From 83629cb7d7c4ad9f4417299c00b913fe6c824528 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sun, 29 Jan 2023 19:31:14 +0900
Subject: [PATCH] from local

---
 .../.conformer.py.swp                         | Bin 94208 -> 94208 bytes
 .../pruned_transducer_stateless5/conformer.py |   4 +++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/.conformer.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless5/.conformer.py.swp
index e8ba2235109bfabd953a487d6d987996acc13f16..6d87c6d76d2f4ef3a84f88bc7dadaab8c87d2ada 100644
GIT binary patch
delta 106
zcmZp8z}oPDRV>LM%+puFQqO<^2m}}yLcOjfvzBZWyCuLVyZNJlus-i5ZUzP?9*C^$
zWI=~)-lafZGZ3c$vD4;Ghn?J$Ykl;XQgbG+_K}?2>Z8qKYpY<?yvt|%E+0lWWdNFp
B93B7w

delta 119
zcmZp8z}oPDRV>LM%+puFQqO<^2m}}yg1oLJ|1aJsc1wU!dh<sCVSV1O+zbr0JP=vw
z$$}2qyz_xPkVp;?yKU}t*vZXOQczkl{e2#z)Z|I`L?<t}$3D5q$Al|0Pa)b!$5h86
Owt1J&_FX=VZpr}D(j$Zb

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
index 2b76b3dbc..e731ded4f 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
@@ -696,7 +696,9 @@ class ConformerEncoder(nn.Module):
         residual = None
 
         for i, mod in enumerate(self.layers):
-            if i in [2,5,8]:
+            if i == 2:
+                residual = output
+            elif i in [2,5,8]:
                 residual = output
                 output = mod(
                     output,