From 88f0e738b91f13e4714f398d29bef202c85aeaa3 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Tue, 3 Jan 2023 22:42:11 +0900
Subject: [PATCH] from local

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

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/.conformer.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless5/.conformer.py.swp
index 643b6418829e2ffe176f9ea51e4c16025fb787fa..1d6a221cf102f6195ae455b2064a89184d464de6 100644
GIT binary patch
delta 855
zcmZY6T}V@59DwovaZZ<W?hIV6$abKh4NbPuMco8h5fOt{Ar;2bwJ6e*(}uP_Hi+t?
zi?SgLgAxiP(85)Y3hJg0G8YOh3Zks+$`p|x=)(0Js;gf3@qQf6c{x0(NpmVy=4|yf
zZ4GMc8hpY~MKZp7k*CAX8N0GLZBm0ubvmThIBJZijO~9W8NItlJ#WuyrE|75Hj#c9
zFIQ(As<Ja_Y`bt7Jve~~Rzt<RN|8~VLL1x|w~Cy>7MSp?LgWEju@ZBOM8@z6*KrOp
zY=al&_+b$#SR^UWSq$MSbTmSP4YTDUpHRRX3}FyGxP%142w^RjV!BM^A^MO&GXij;
zU>13aN4Sl!d6P(x#X_mbM~oqXI%97Vxq@AAVWC8102k4Sd4)g6ppyIGP$sPD$!3?M
zyP-3xx5r~qkH22?)oTIc)cfD5S3B6DYu(!6qgq?b@u(i|h{rnPv3<GEwTgR3vgLrt
z4X!A1t<maV%A#H58>a9YBS;~FI@DskO5_Q2tiW3$_Mr(bRN_7nyRh+Z(B;hYgnW+|
zR-#;EaT0rBRQ6yA1hW)2iud9e_G1GA@L&}LMJi9>EAq(VCN7{I>tVwTq4T(dvp8hg
z&4*YlLlqQcDccXDv>$HdDS8@yOcQYcM-aj<0)9a+Hp7LYX?WsRN#?PlROG%pl%Lk$
Dx3iHk

delta 429
zcmXBQKS%;`7{>AU=fBL$yuhPLFo=kty1>Pu9D*pq=npxxM7lIpL&GiMi9-=I)FD<=
zG)0qUAVEt**%UQ229m9z%?9<2UiiQ>Jj2Ul+djMQHPU8kCK-*7n8N8ImU*2yEgPR9
zWjJm<snX7Z-dowzt<YgTdKLAwf)f<q)qgu(?9tUy_Aw|@Lm4N?U<%z3RJugYu?!z>
zG?5*|(GG|-af}rVzy*cZPLW$&B8OFsV+b0){UQy&WaWi$0ShTiq8|n{1kmW<Z`AOB
z5(?PG2IkR+mQUmvB^+TLv(WJ475Tyk3dkeoO^AdEE|feXYY5}XEwYa}#PH@4xrc>8
nv>EYl$fh%vi#ca^h7+HT6c0AlaB(Z4MyvL=vbAa>q9TjGIvYy^

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
index 11ad6c64d..8d766dd37 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
@@ -197,6 +197,8 @@ class Conformer(EncoderInterface):
 
         x = x.permute(1, 0, 2)  # (T, N, C) ->(N, T, C)
 
+        layer_output = [x.permute(1, 0, 2) for x in layer_output]
+
         return x, lengths
 
     @torch.jit.export