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

---
 .../ASR/incremental_transf/.conformer.py.swp  | Bin 102400 -> 102400 bytes
 .../ASR/incremental_transf/conformer.py       |   5 ++++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/incremental_transf/.conformer.py.swp b/egs/librispeech/ASR/incremental_transf/.conformer.py.swp
index 346de450b5d70ff3b3a6b641075e5735eeb77f43..1022b3fd465710df464b5781422f793ab4b03d38 100644
GIT binary patch
delta 528
zcmZY6JuE{}6u|LweN?M0&rpIUo<tLaugO4y2rEM+Cc*HuM6~)|6C%=v#af0YqzQ>p
z;%j(|g_tU}5UY<#NU&K@|As;0Ccm6>b8>Ufy(hWtNNzh0qMEANnAaNA@M=o4Q<^d>
zy*ZD_pjSjRk(zXsu7chS+a!V*wC8no?b&ENuIr5%Ry;PRFSO{h#$3n>n<SHqMr=G1
zx1w>&6#aKiPw64U8vk4D5=lo|Rfm1qsyh1<yG0@&#Xq+W6AJ60Qsf>NIL0z2(2P3d
zDnw2YfsQNMOaHd%Bf4<JYb^A^k9xeiMQ(78EiAevA)_P%2%s9D<s#2G!#biELl4>!
zKodS)G>&`RA&Y(NU=1_qMme6G)Uk&Z7zm)&IU?dF`6}ZuG8jS|{Kz^)me7XBQjuMR
qF@*OLkz1^zA3hX*GGBanx>=NdGnL2Ao9e~wD-7e`$n1PVDeW8NUui!8

delta 474
zcmXBQPbkA-7{Kx8-I%e>e(W-ZO^St;IXEgL*DZ2zQd(PK?2qgqTgge(T5H*1R+J-F
z{El*v)8rs>_2(p#qsz~CdF#{rJUu-<PfxaD$yO|9YnpQ9<C;eWw{x1F(^S&w7<Gu0
zA)1J**ydF;j#biq>|sNXd)0%zc#u&8#j;-w>xX_7p2!@TL=5wPTad5*45!FD-f)LY
zWU+`rbm6{L<O&J&;*l98%)t#iF6_+3Bs}QGk4@wmw<y~rBbx;Cn1&A>5PY?W+@OFY
z7BG$x_|Ol5(afTFMGaLPV;3tJg9D#d>Nv#?qVQQ~MLG#up-^oSsUU;^Jg8Yjwh+Ka
zqsR%C5W;VR$O{UXf*XeU{JLZ+ekYaP6p7}`Pknl=N2P7`(}&s<;naE}=1mWIm&370
MDw-7C$S9@#0pF%oxBvhE

diff --git a/egs/librispeech/ASR/incremental_transf/conformer.py b/egs/librispeech/ASR/incremental_transf/conformer.py
index 252cb2e3c..822bc7a27 100644
--- a/egs/librispeech/ASR/incremental_transf/conformer.py
+++ b/egs/librispeech/ASR/incremental_transf/conformer.py
@@ -198,7 +198,10 @@ class Conformer(EncoderInterface):
         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
+        if get_layer_output:
+            return x, lengths, layer_outputs
+        else:
+            return x, lengths
 
     @torch.jit.export
     def get_init_state(