From 7a2448d887e0968ff88b2b499bb3bdeec4927e8e Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Mon, 9 Jan 2023 19:26:13 +0900
Subject: [PATCH] from local

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

diff --git a/egs/librispeech/ASR/incremental_transf/.conformer.py.swp b/egs/librispeech/ASR/incremental_transf/.conformer.py.swp
index 801809c952c4772cca82edfcf154cdcd4684b285..ef2213610068d72fbab5c933df6affdc453976f4 100644
GIT binary patch
delta 935
zcmb8tUr3Wt7{~GFb<Xxj-Io87yf_+B!dgp2>LM>zAQF;7gtAghje>M*>6#F2c(E?b
zA}kjL5e!PZN+K5oiv^`cK_l%Vqk=AiRx*Qbgj(O1^uo&yd^peHeb0H%%kztTHbx=^
zWyjoyn_JW#ms=Qik!a>|&-v-HA2zLg++c6fO2*sl6^;u1DWmn@BrhIx*e}>d)P&OV
ziyDeVZb9E(5_2f+U|3)Ja0z2Li!Rin5c7p1&oPV>*o;?JkudflXAw!^Av&S(-7GSP
zS=_`3{Aj{f7?3rIq)if*rxa6&VE~P&f*Id7h%6$FH<-dC#&8iqc+rA-l;dlG$P6YB
zL_1t?AZ-+Rh6L`xYuqo=OtF$L@*Z;tVw=7;h+IW0O0kkB62~YSv83_mnA8%B)!J)o
zSgnpz-NSu>F5l3ge`xTG+N-=Fm%lI2KQ!34eV0?YovN-@9rp$P_w|rEIiS{^eN@)Y
z2=ST3(hhB-C*0l3wYgrFE9-q;3qsB&@(v4lg1b14L#RSAQpF<o(SuUV5`a-O!HnND
z|ITU<>Y44Z;R``{iAPo%dYPgh2e2EJuwa>Fe8LQ(=tUdqumzRagk?P)Sin4zh~pa0
zp%Xjthk$%U61NdVhpC-|D2k9H4L=db6?jksJCdZqhZ=k!0M~H@9%N|#TSU<SGqQ&1
W)Q!AYQqwGnOttpQ>Ts^_g~(r1ai@g<

delta 430
zcmXBQKS%;`7{>AUm!9dVbfMCE5E2cEVM9R;6%8dKK?N=$*U;3|U_;>L5H*U@1uZ#6
zll>VCT+61&p|K%tjMCX4`$jK(;2oaf;jwLx{ia2%_{MU|oQcJSGej~U^{wP}<a0<x
zN-iU%tkSA6a(iU#4HcVa!8|dQJ2*$QrT+7lGi<7Kz7-UCL<3hig@q*q(GQ5+pn^4M
zxbutbV;VhO<O>cmh(bZ|;uERk5;it5i*e}q_DWvbUXdm)ki$9_Fp4mA_|P7tI~sUE
z6=fV^2Wgn-ctoC2MG@OrfsP+dq>FcyQP37e5<DR&RNW$3gz)4NIYtTzybg%mBZo2c
i81iq-PHFOBs@M$Fy1T0HcPe2ObtaPPtact#+4)~|eM^x5

diff --git a/egs/librispeech/ASR/incremental_transf/conformer.py b/egs/librispeech/ASR/incremental_transf/conformer.py
index ac3a91ba2..1006fd0d8 100644
--- a/egs/librispeech/ASR/incremental_transf/conformer.py
+++ b/egs/librispeech/ASR/incremental_transf/conformer.py
@@ -196,8 +196,9 @@ class Conformer(EncoderInterface):
             )  # (T, N, C)
 
         x = x.permute(1, 0, 2)  # (T, N, C) ->(N, T, C)
+        layer_outputs = [x.permute(1, 0, 2) for x in layer_outputs]
 
-        return x, lengths
+        return x, lengths, layer_outputs
 
     @torch.jit.export
     def get_init_state(