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

---
 .../ASR/incremental_transf/.model.py.swp      | Bin 24576 -> 24576 bytes
 .../ASR/incremental_transf/model.py           |   6 +++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/incremental_transf/.model.py.swp b/egs/librispeech/ASR/incremental_transf/.model.py.swp
index 93d1042ecc5e1c2993f5d659d9d8bed825dfe319..b0d71b1df1e880f20d493652d01267b95ff3d208 100644
GIT binary patch
delta 584
zcmZwDze`(D6u|M5phlD0JP;FfkUKg&q8J?nL-9w32>uv?4uT=tr-(6$$qVXGlSLQJ
z^!%~7xat&RJLq8Op#Ol2v>;9nW^yXDP`@@&q<G-NJ=}Bdy@#7C*>WXY^GxSNXhik;
zI)%_gcFO0og+lX`W*aXqYgR}0mt7=WQz?!=HxU<Zku~VJ`6zOXO>7{9Hk2Dh9`GG|
z*hLgxcyRwg<Qu+X3kh_?4+l<N4C9ilm?UWg;e`ixPLTtA!8H2d#(e|B;Q%WbMmt*3
zg2#H1D;yz@I7Tpt0l0AO5IMvmdK_U6lW1_`XPw9oY-0$2YDIqI68lJE6dkbRmY+L8
z9$Bm-TvMFrFWaqUe`|3eIO(*`x*DyV-}TS&r>57uy5~{zh8c-P*Njvokuj4QGZ0Q?
z46D%P`LEUCUWulbGRZ(?(baoBq18yth^KWWsv1UoHu1S~xvpNSZ{JI+fJz&&IbYI@
PRGZsZbyfU3X49SkAi#dz

delta 462
zcmYk&&nv@m9LMqZ*l%OsgKsHQY-0yvTRAAjw99JpBTf0Pgq)Pn4r;V1F5~2LVK?Kl
zi_#91GCvMd%7H(C8?~GqIPqK@yz5b~-tX7xoxL<=FHQPvFcBMW?duK-qlu&|=W|;p
zx}llkrDaX^WOTF0jOF(jua=lIw}`?n66nNNgUBZyaEJmH;fD_&^&&U8#vaxX!5|uO
z<q|o-2C^83-z6FG5#H-WipXIBF?jId6uCzco0vo|0_ecEL*xmk$Rmv@jA8^$c(IEd
zV--UPpamYh*(O9@Z6f;^g9COvT1Aeq4g>vgqr#`k$YTfFsD97D6x!jploF9jjk*if
zsoa29U4*UXmHAYwDu;9yZ7Lb{{L5YHHdOytbg1XBxs;4fI#tGKu3bt^tGwYYjA}pS
CS5=q*

diff --git a/egs/librispeech/ASR/incremental_transf/model.py b/egs/librispeech/ASR/incremental_transf/model.py
index 536e15a9a..7c0138405 100644
--- a/egs/librispeech/ASR/incremental_transf/model.py
+++ b/egs/librispeech/ASR/incremental_transf/model.py
@@ -243,7 +243,11 @@ class Interformer(nn.Module):
         x: torch.Tensor,
         x_lens: torch.Tensor,
     ):
-        encoder_out, x_lens = self.pt_encoder(x, x_lens, warmup=warmup)
+        encoder_out, x_lens, layer_outputs = self.pt_encoder(x, 
+                                                             x_lens, 
+                                                             warmup=warmup, 
+                                                             get_layer_output=True
+                                            )
         assert torch.all(x_lens > 0)
 
         # Now for the decoder, i.e., the prediction network