From c19081f271429d508125fee7c3aa3a5b2074abe9 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 14 Feb 2023 14:42:54 +0900 Subject: [PATCH] from local --- .../ASR/conformer_ctc3/.transformer.py.swp | Bin 0 -> 57344 bytes egs/tedlium2/ASR/conformer_ctc3/transformer.py | 4 +++- 2 files changed, 3 insertions(+), 1 deletion(-) create mode 100644 egs/tedlium2/ASR/conformer_ctc3/.transformer.py.swp diff --git a/egs/tedlium2/ASR/conformer_ctc3/.transformer.py.swp b/egs/tedlium2/ASR/conformer_ctc3/.transformer.py.swp new file mode 100644 index 0000000000000000000000000000000000000000..a2a9ad29c0fb1d87f41ca2e42226b9f4b3e38723 GIT binary patch literal 57344 zcmeI537BM6mFL5jR#3!t!*+jEsYF&~MlMwaMMBUdRfQ%NRVA6lmJ~^3#LLW}BH{({ zBC0YAN7PS55C_@DZ`5vZ0r$P9h5od4#I^@*#a6VjaYRQ)0T*xq6_=jhxy!rvMMP#L z3XEvPSO1EP_wIV`cFwuyo;x#i%J@maQMsWEpTEjvg8E&>b<^t)xOX&@DMj<)(7N>_ z@@l!yX2ON7^1w-j;_*?mdq*S6Ria{{(jMpKHo?KYQDb*pFBHSUaxt7OR4RkvTs+te zOOs?A1?YS@@AHjC>9n~g#()(>*^#rBkYzk3wuQD9jLw3@R68=sI_ zH!`e-`s!7O1y6bM&Sj(YUiK)^qd<=WJqq+F(4#<)0zC@!DA1$8pB4q0^@B6#Q>DK! z)he6sN40$)V18SP;heVje`$U{*?fPh?fnzX?=#K!&28@=V}74uzSo)p*m6A9Jl}1; zFKByzpm~0-`M#&^{Xyn=&3qS3g>CweGtZmm`{uUyk2lX}%=e~8>pyJ#gU$1X`M$R8 zz4@6LH1D5o2x{Y-$Gxu}1$q?dQJ_bG9tC<7=ux0YfgS~V6zEZ)M}Zy%9uf)^N|}tv z=3fv&l>L7z|9|j_napY66!1jw1n~JoGMU$cv%r~PE$9Q^A%UyGBsc{82x0h#;9KAp za4mQTcq4cvI1Ow7j{)}}SleB!V&o3qjCQB0&AQCx18qgtV&;*@K1S(}lqT;CE2gw>g_L=u~#+Mbcp zL~AB4RO=N<5tq+<;C z1+mLyG?BUSVM zmBmpnR#U=+K|Eimhe38*fA?%$)pB0k6p=NJFvr;wHkxEkqUBn>)eNFpKlOf}03N*q zK8->-4ufq`a~$f1)v$)8&z43bYGhYUwCeS!L1MMaVsLmT9rEiB&#iKFFO#PuprKW8 zSUyaI%PB~-pk)TGiXlHY`RzdoJ8Y~P^s_+7>8XRSINw=Xw6U9Odz)6-d3iB+ps zB?&^EpcFR4Mzvg%riHqN)(*_%z%da z&Crjl-rAWV9nnT9*KFQ{rBDq~KLHae9Th6fgbnEwCPtT13#H6kUHW*bG?uQpq+!Ru zrb{{vp&&D!bWkR%uog2jNU7wxHpH`8oqDtcn-X!ov(*b(ef8EedW9ny}{KAEGpMYYfc>==EtWs{j;om<(`gDjY> zL`>q+C5DdN;6mAvtuZ@UZ^lhG0BKyBl}1R5T3q1x8%Q4?fjX#GYPNcT4LYQ1}I z`;?8e!OOl&+jb`4x0$?kNz0q0A=0m8M%DpK;cSphqZ!rsKBtha)p93ArB)^E!)P_b ziV`;Vlx5rtcd)r!Y)-|^Mt{JnTAa^KG9^TfU76Lg9)&Ey@^qruDA$`oBRscNX2DmH z2Dg^u=3v+?=7=fdW42XcjcP8|!%{AjnQcT>lO*}Kn&nEIYt(8%xvJK)dZ9EOlN?PG z>aZTAY@iOdBcm6I35;jM(pFZ4LPL{4KiMMeif|kH;pV6{Qwb;bmg9N-;P^sZE^fnW zR{tC;Hp_dMpQ76Fg-U^fkeDiVO!}l^i;>k&D}|Y`G96c=NHG`o?N)g_VV`b|;#dbS zM%5Wg*s$SFF4Rg&3cPW@*;` zK$@sO*k0G{A3RPA{=!DC7Mk<+?_wlI!>fZzxms>2lEgu)2J5NTg{@d2*J2jQ)xks< z`i}w_U-dh#hsE-2xyZUvn`;&3!eB0fA*xA{=x^0>tUDY;AeB?cCttjM=VWl|*vTi4 zZJQk5G7)S)IoPy)+vf4f@$K9AbwV(nu!r8?e;$vmlOUn{fVNW;N-^XTI62j}8v*-Mcqez=C@|%tej4 zLDCWm#)DhOH*MKAv1MQci*ku_dejQ4g~cG6X%dK%IRR0o2bbBILJ0n{>1&x0*%&PR<{CA5i(Hp+AJO`+Pz z(8x#-oK`MG)iTmXqn?weOMkLDrP1J}Eg4qiaVq_;k2rwWx66lkRt3RB!dw8 zjTH-}u&N$S6w5LX#X*wni|Hn3%(wb7BLAO&y!>2bY?1#5`I65sk@r6a-UF@zvtSgg z2hRtq!7A`|Yp-LN)i5F9rLqS*oYW?Nw|!Y4u&vYhDg1jo}pNeEGd8ij2A6g@rl zSh|=hN2#zx%LQBf(y})p!lf>8;OL@@p|yG$DLf<^tJkSqpvy^6U2iXcX{SP|G);(X zLg7)RX_!hHi553C6`hr9Ko_9B8K+{?fNQ0)D{sb$l{I76R*ahkc5GD5_tTD&OR61p zX&?M{e6X5vMx0{obXJKg?RYfH$e~apWxL>Pu zvnYJclpr0jYuF(hcGW<+HoM9TE~({|>e%ca>Ewf~DuVr?Rc9!PT2xe2oan0KY`hQU z&Az@qRhqt>IzP{TX_zIu;bf9h^8agOPnmsZk^ilI|9!~zw}SVBSA)~RG2jK@m&o=1 z3a$ep?_U7U1N*?qKxF?fBGZ2c+z74#i{Lb{3LFORMwWjYxDxCD+rcxzuaMop0p1HP z1}_8WfvwMQqOuaTT#R0B z4z*ZoCWoSIurePt<8T1YWmudyzXoS2(aaz{hxyh%DixZAIBX8;fYS{&{N~uQL~V?x zh1FZvp}1Nd)S`0>8JnSedun!BjC5u)uc3CetAT`zJ$1aBPvng(C2D7e_-xnZf^oK& z^?o&)KiRUL=nppa`H3yK6ufC&5@TvVxy71Dnx3l&ySJy)Q1i4>5l`uXqSs%iKu$g0 zwaaCO2DI9$k0#qR!?_`?VwbE_HB6MzQyzihU2H_i7Ltp!kmdp0i}bcfqgAuJomsaC z8a(X6B#L|d=@B3tpBwCXWP@zTUUD1j}m2!@Uy)VoNZtDJx!u5m9yOwuv_C@I&P}h2EjnEMK_e)nl_DtYMdFK_(2gah&?70 z-)se2PR5pfPodIc|H*7c*laWQwS0Hvp!DO$`h#)AXksXJ<`mFX;J`|4iEWzPWQ!`< z;(|qQ>(il@8mC?g3Ad^G0a|;-pNXb?UBE6k`)}Sps}9TqF9BiV0>ySSL9nsMrbvzE zmhH4=8Fw*@>84`DV`NdAJ6d>Iw%}A#AImC?l7BB+dDz=kdiPHHnrx)hS1DT1BZT6q zoC|uCOKIzo?CWakH^mBPu2CqpB!8x?Vk2bQWQ#&J0*9D+8c}O*o?+vqX66m=Ne~c0 z_R;%kZb*xTj)iUyHo{ViZRMzzUsYp4!!#=+N!U-i2g}pM z6fc!aViCYFC>P@VWTO>kY$EcV3a!_o#7fZL9&aN-Xa0$;Mx3A8<(65SmlQ;s5#{Ai zOFjLlGkEHKKJ_ad*e^dV;2Ab<)&^DkEx54_$(!$V@>2XZn~vpeGbMtu@91WqsZv|Y zMpM1iWlmb4vpQSb>@+2oPlM?PV5=!{?WeOrrI`}5%Q&CP#Oqc`kZI^-El)ExS@}-I z)K-r9pv};ty_JL&p246}^R0ynYd~o+n8#=VIdTx4KoBohXQE0fqa?GWcwu^lPn9GI zM7Pm`p2UY|kr$~R7YD}%L*BZXrWIBxo3-_Cwr)F7A3U{ZTjK0wlT6jZWJjb+P?gYf zW^Kcfj3$z$*_C*8Fu8s6_UtVC*5|ZpeWNJSs+dezlUz4%xLl;7eVrXW9Y6xfCmGnr z1hO-pgp!Kca_d12tK`{ONf`U-5*FJw=L{&8xx|+)4q%?6}G-&CUlCU?)q|1{o zcj{d}^~(kHTO&QWY|(6{%YL<2GR>+C53bADis=3SLFDl(fXM&${QiF;=YIvf7F-A# zpafn54h6qL#=jA~8N3Nx0?r2~f)Q{aP-#AJrE4!nj{-dk^eE7yK#u}F3iK$@qd<=W zJqq+F@aIASE1b#qBSmNH>GP%7y{$cW)F>^s>tAAI(tYE~Z0%*)BvTrLX}b^id+zng z78$!YCYkoVI?kMzGMQ&Evi?@~m689?Fr)0N$oRK_%Rm(j0r3NHKXU!;;8WltuoVo0 zXM(SwL%0pR z;6U&-_V_;o-Ui+PYT!(82>74u?aO)pE5NJ3-++4#%w)a`-UxPsjo{bp>E8%8frG*0 z!PnW(e;0Tu*a4mc?qW~>DsUxefuq2$*~`Bl$Ql0Yz)tXZ@NxF;tKfJr27bq0{eOVB zfb+mCcnRnS_d;vg#}}HvMd-fNM~0I9=&hu@{&IavUN5mvnP1|0I&yQa*|A-m@Aj}W znrvRGpz=rhwa(gdrY zHcFbZ_1VC#Ih|FQwDR&{K>dx*6X} z6#1;~W{9rF_!^Po^>xYJhju=`$ykTZ|{cy1u7KVn0H_VpS&&&=L zN=J;WTOUdU)#U~`XFHgjrNuxiE;2rH#n$SrBO8y%sEo~j`T%PeV=z~SI>(p?Hy%0E z7H?9HTh53Pu9}9zg|Mj3oMOSO3K8og($I3#+O=-(VPA@))-`Mom|&ct<zdkIcFD7e#GHjSTa}G$LN&3PXzmR$^3caa zO$252V5b$cIZmur1q*C9?b_XcT@Wltp`;j;VCf+6W{jPh?3yhG75S0XmK8Nl zdUBMS=`fnZYI=8ADbGhyDP`m{7K{vRHr6d-QhTFaOoVzn}=(zK}!nC!n$kni4|BzDycs@ z2bbV7E-?Jj>>?Nor|y=KrU_Y6Ja*5tQTm~yp0>32GxMxAmLf=Ms&Sm9sofH9g(cKs zMI5uUM-40Pj1cEIT?RSv!O47l6uEWT2x&HoOcm~>|CGfe#ds!0Ia$l}_M633@t7_V z`?@^=-Nt$|IZ4CurivrgEgJ-08urib476pL#nFl3reSPtGCFk|CV4Ecdos!LGRdYt zevw4v|ED5P6P*m-y-+_27DWQ3%nPc4W>aBd;=Z8ZQvu|Lhu@J zB6v1<7Ptp}z{i1{12`9mj^IG>J9GlS1z!R;gMDBd5MO`yp%=Ipd=A_Mt_SY{mx8mv zVc-DpBlHE|1|I|;0M~%az-BN6?m}nqIdDFxf;HgBwAYV-wA*dcc3nPp1b62wW)|x> zt5ckh(llYygf;4{I0;d{q5>qP*#Zt*@|e-4mZd<6uZ^gJ_!?G1@iG%~(oH1?=c6rM zjuw-pW5fg(;a1#rR4U%wVU`W=2`o3OmXhrhlg*E~Q55s!SRAZqNvD0W46>2&{o>{; z47(b|4oTdtSVEPqSy^{IUM4J=6MbGL{(>*LNO3z;P7)F$W^XChWOlrvU62E;DwX4* zG)GAWF73AC4m)hq8Piy8eB7CvE3jpL)s*BfuP%>e*Yd7Jb!M%8ik+t&5?Vcg2rD}v zVVM1_TRQcEt1aJHK0PxyV@cWP!uq|2UOmszPFx^~!QcQYCS3A~s)(a7d&2xsZv6($ zzI%^0m`8<~IDZ5lZ=`R@N0nj5I9;L;ot~I9k0D`Kb6`beEEvKVZ2-g84Zb0Q0(fCx{7ben2{pNXLT@c{Z?wvKMdg$oKsY744P3v`A?-R|4k#Zi{_!z z>6L8(d}fL>EUzI{WUcr&y_Ez~ztj2XSywj!A^o4OFG$I>3(iyIAc^t3nMxIL)kw>U zfX&H8B~~+oryX#nBOW6@?Zkgn`th+c^Lw_b-OJ#Y_kk246)!14>S=oc)PbM0KhA`d zc5akH%10*cdLx=C%;1Ohxa;uv>#xjw*Ap_(uhX*ZZYdD8ThTA+{&H?YDW61lGq$X@C=e=D20bzU|54{0-yfWy?f(x#cDz$$NaTNef8OVi_pb)80K34M zUfz}rC!oCppD-$UmAJa`Y70jGge!5Yv9ZbAN+y8+Gs zJHU4Ee6Sik85{tlJ@4@K0=6r%C)<;(WTU+&Sy`{Z*@s)QvVC!7f>kn$s;N^{2`d+@ zY)(3APbljNW$HeZG<*IC70TFP?@47`E8_&@KR>BV&5cT>`)4YaS#Dm*p<>i$ec(7b zL&P&)q{Xw5rNlG6XPgeaq%4=pU8JB|&Xc<~>r?X&Oq@eB+rLETr=07Q8H~XuFX~FlSu6J=onnYlU`wTa%%N; z%IK{6ZPnvSmhN+D;LS3J7m;eVFE|Eg1T-z(Ks7^3?bDf?-bvDG=~N!J_)dZ*N0lNLVakV( zTbjj6R`eXr5GANl8Dfl)q{q>A)lIdZAK4ng8IQW71D`&ujqvzJ*P<1gnBs~*Af@M) z&`CmmX`w&cKBk8%?Wa+$uQk>Vc+X=y)HRBKSVD7GWa;S8k`1( z!D{d{VE6d{13CRBa6OQ_{VoLJ+wVo-IpEnq&H&s6z6L%Gt_EdrCKv(3K;-*h0J-z; zGvK|T0c8LGEU*il362ET4&ZL&`fI@i=mWn)p0{=Y-{J4Cf{%hXf%8BGoDEI}M}Z$B z-`@(}0QP|wfMzIT5j@_)=4Y&v2SnB#c1~JEZ04CN0xNz>?B*e zT{}m3yh}N&xSGkqaWblV5`@1AZH|(Tyx@+sZ6m?h1AK`|CaI&ddcTH>S%C--z$30f6 z&hp4&!)9KyP~0t@W1G!ZucUD&%eloko=HZSx{J%sF^1cO3wP6j>fwql*UR`C#iVy~ zn?zbEx2d^&J{?wnUM7E14?1L|FcTB+_H7xuG^A zw35#wgRz!&GmJHVW>m@f??Gp*CoN-*xAZv<(JjMRKS(^dwWYQ^xUJfQZKrAN`NaR~ zA7riB#b8p;Dl0e7^d%iLeYMX5u7e6elyQWd^t0BWbb@8Gn_S9JQ##TE%*)g+=?cQ+ z>5G(UyCp3`O})Ash04=ZUAiZ}Y%RK{oFt?T0MH4;tYIH0Y=GtL4A-EzRb7$&7@z98 zbT&zn8Jw*&kg2pkS>Ll>@3wDB&z*aB{`oZJCPtXZ`8Hn$IkAZiCH-If* z5c~i=z*oTqU=b_;(GhF}KSdXC2e=V@8oUl%4CL;BRp2lneg(b&J_s%c`#>E;K+X&7 z02{#5f!GLq1hl{gAo_tvf$M3rk#MPb!0(5+y@)y&<;5C)&` zZp%QV(lWLNQAt>dkl8>vqtVJnt-1AFUKmNB$nrOVYd{E|55%7T7s&1Rfqw(<0b&#IO7IGB4k&|pAb0?8;-7AKbaX^cZ0kW(#5JBi;LI7mH5{PbGtD(Z z)%Hk3sYp10T0WACE(Kc?3~*6qb3TVVCC*9l!J;M@)SQPo_LPpH%Z8IExpxE0J;hL% zkoW;iJ7`lr&(?XTu52ZeN^(Zdt>j5aNWGb2g(AH8*&@gK?GDkYVb`AfN+Q*P=Oaw=iUtUTT2`E6X@`H5J& zD>{d%WWbGSNm<=W;Q|UHDs~}hQ&ZWH@e^KwsndRqo$&Ou@?Dug?nhJm)z0F{XrWmS zQP{8a4oOtn#{G@$gt=fl>uzIjIS!EnB5ca=6-HJiuD)_i0S`Ljjn7hrV}nUfosmLK zTwus4i#dJ1Ul&hSZY&gV%99Xql2tx=dA50bwqz+jFC{fsv{YC!G3>!Yk{B(sWdWM# zyvZ6IA@lUO8HE?p#hP*~Xk zZSJ@mDIIu0pJc`yP-d$5+(uLL_LlzX(B>({Gk>|)a{p=i+<9O=>5-EhfHfuSIDyV# z>w3K+Cq2#Hi;*9ofiiYP7U8_6>z2vgYooyy&K|4VK*b{hnt#q$yKvn#3Mq>jY0;$X zsl&JY(gbvuqRR1faUrFr2StE%4Q;(s-<+&vHXbRfRUugrWI3>uPA>(i$ohv)Vqkbp zKr%@@5=@eFcQ6kVK(9Gxk?CNP2Q4S%7)YX|U-xOLZCli|$8II-9b07CV*x6!rRsU? zRn_9zC5uYvdEj)>B$8P?mx`2;j0S=+Zb-*fA&c^O9E?>e!Rb*e*oG5|(Zu6qp|Mcj zlZzU2gZ0wv;P8<{cqCbOhKUw|55_^R!^S3O`*=@6M1wNN0s<%~W^Xsn4 zu=|5cIB0fhrL`UMT+b`%=ZUW3q0@p;;*4Z3ADK8Z>M!-*3V4eBm`^_de&P{kend&8bmBO3-Iaqv% z{Et|w&x*7E-!Nk2Cy@Cs0IvpHft&$&9(XSJEAV~f|L=jXf@{G(AbNp~;7G6*JPCXc zUBJJ9&w_sj?*&z`13V2p6+8}n4LyMP{u6tE*MOISD%b=L0r#U1xCIo!)4@-X_kRMu z3q&vQDzFdi0^{JtAOME~IX57C{%;0v02hD=)WEaA_mTbI3TA=q_5Tcc{~y5(Ky(2i z=mXzJ=Km157#st#;1|gJ_kr(${|mkgE(Nn71V@5{z=1%{0^9>W46XsQAOxp@XMv}H zpP&=?Civgr8{qH2PViXp74!mM2JZydfmeePI00+{M}xz`3&AtM9iHw$&!uC5c9yOO zniH22t}w{P`w*}9X`ZWHw;S&p)iFh4%t(@LYDAq&5KIl_z;a5<6muCPSyj3icx7 zp+3NpTKTBgMUrC3WM}7^oN{a2ErX0O;rsJ*FO7o5d>6wfy@G1Wh?`l-LQ&HE4fDxg zI8X`l`NY?Xz6{-G+{c2MD5`j_tySUWiUK`D;=@~>80ECGE7Bxr?>>5t*G^rMOUknn z*Lp}sJBgE|KH*Pw$y9NjBbbt##U%=@)wK_6ulUkQ#Yz0yy0zLv)-wT^r#pRS9U8&s&(h?vM>!KCI#<`?g$ve~le}egAwm9wcRMg!i4p95zUh+s)ys&qTHLYAo^SmZ#0_I zvke)jYn3^S`rEmDldS;#NJ5L6Vh^tTqZ+qN%%LP`##5w(aJ$Q#0aaxU`1mO=7jw8X zrdP&gGac%Px?H0dM2r=!NUK(t6Dzv2Fh?^pwKzel>#S&4S`aE~MRK<~Yh@_<0zEl+ zO*5nz(`gr78nasx`qGyi9RjtFK0ez=3k-8AO_5ODLXz|kUs3IOxbh5?6vY7Oj%rqlPpp5*Mb=oVutk_O}?3Njs zD49Rd_;IFAmISjNa?Fm~T4Fup2Ej=(XX)k@-Ji#|EIH?EIks~x!e1IggCRZ>(lc1h zxIaexs-d}XQ(Jb^*v`rE?b}90$1HaA98!+E+=|0GlMCT@6vdTGN4=C$NJRyq#KV4< z+k~ULnmXpACMr`Zi%QcH)NdQNxcarG!K6L(NC*nEYzO(YvI$AsER?JGp)?_B9?cS1 zM}#({nJGI~Ursp1__0(Dow02p9~%Sy$u&+e%he&UP^7;;r*1ydu1n4J^3C~zRU;@h zmi6HltXZRy592kQG$WAiWsx`#L6fYf2zBvMvRt23*loB-)8kAB5QUj{{L^qEayv;b zqn6+8#woSAVMnh-yL5$Z?Mb0Ye{oF|dmGwb%f;54vf>TQPQ35};vH1;24;bmK4wOr z(!d)9ZFa$R=LLrC0&|hk(929U>h^s2;mj6Q5U2e25|Kfhk<9exVb#ryd4peirMfB0 z2Jt@)l-;&o!&8jeBL|k*XxsQX57@(Y}z`!o
n`i~*73p9+3} ztS)~2o1h40z*evhyZ}5F+=~4E_uy)9F{px5z=_}(um=1PS^g$)F=zpi?e9XCzZHBE zydPW%UIo^J?<2eaBe(&)3!DvdKf@3i~z0E;O{;wl!;Db>Kvpt}BHZc#yan8T$ft^H{V8Hmik# z3Q-0AH)BiDZAn9~f=M9iFEK}D3z$WqX0%JKfAYpzi!wIqLd9l0_R zE=xNp9)fo26dy!OYG%!T9TzHjaD^1L+b!kvGwyad7#B-*3)<~Yut}P3d^3BTSnK1& zJf!^~cYR(=3*F{D~}FXzfeBI;u#f@^6ku1!vK2AJZ#Uyach`YK0PhDcIs|G&_xj?4JsddFj;9? z8MX90$F%+amxhT`mnr(-E>9@dW$iE*Z&RdYRI&KaRL#h?nn80#6aGy95;o83=`6da z5;;z@cFWuv=InEX8&7+g#`<77j%J&pJ3#ewz>2zw-c%By%aCs{X#DS(ksmE zQ_?>=B5Lkl`|Oa#MCnMC-;g#&-x5ZQthnw+ZHXC%Ii0Gt1P=<9MOSk`yZ(LC?p}QGF>mv&3vkl>N95Kem}bd`@b1e8Te zbbop<%a(O|GBsNI5TwM0_{gNSGQbywzi@+KW~!1l6KW{6!ZE_N@uW3Cp{ z9?3A7*40oUXVL( z6(eUGz)wI#W^*fdXO~JyEJW4)4i=}xoZaC54S1epB2qk}Zc@u6##|>U6QI{&Q|T?O anoI@r)pQBVVbpKzTq;E$3jO7jlm8EV-I;g* literal 0 HcmV?d00001 diff --git a/egs/tedlium2/ASR/conformer_ctc3/transformer.py b/egs/tedlium2/ASR/conformer_ctc3/transformer.py index 9dbf32e48..189d37572 100644 --- a/egs/tedlium2/ASR/conformer_ctc3/transformer.py +++ b/egs/tedlium2/ASR/conformer_ctc3/transformer.py @@ -186,9 +186,11 @@ class Transformer(nn.Module): encoder_memory, memory_key_padding_mask = self.run_encoder( x, supervision, warmup ) + if type(encoder_memory) == tuple: + (encoder_memory, layer_outputs) = encoder_memory x = self.ctc_output(encoder_memory) - return x, encoder_memory, memory_key_padding_mask + return (x, layer_outputs), encoder_memory, memory_key_padding_mask def run_encoder( self,