From cc5c3eff53cc8c4647da65bfa0b9bb04ed31bfbc Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Mon, 9 Jan 2023 20:21:47 +0900
Subject: [PATCH] from local

---
 .../incremental_transf/.identity_train.py.swp | Bin 65536 -> 65536 bytes
 .../ASR/incremental_transf/.model.py.swp      | Bin 24576 -> 24576 bytes
 .../ASR/incremental_transf/model.py           |  11 ++++++-----
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp
index 33277a7f75df771d00cdfac403a95d19eab79628..3449aace706fed396e086cddd85ce0d6592fffa8 100644
GIT binary patch
delta 33
ncmZo@U}<Pz5lb=%^Ym4))H7fJ0s#hws2{tNqqsMUz4{LTkmCvI

delta 33
ncmZo@U}<Pz5lb=%^Ym4))H7fJ0s#gFxv#sE71=h5z4{LTjR6U4

diff --git a/egs/librispeech/ASR/incremental_transf/.model.py.swp b/egs/librispeech/ASR/incremental_transf/.model.py.swp
index abb97f3cf364ca66fcd8a8f4b7a2b198d31fd765..18b72fcd8f7b0179b5172928db2da0965ce6daef 100644
GIT binary patch
delta 275
zcmZoTz}RqraYKLrqweNFfz9%~ZJZ1YVO$Ih3<3-cx|0PBwljupK55X)DW?Dh<(VZJ
z3MKhP$r*Zi`SIyRi76VIRt%d<tR}G59|dX*<c4UL0a75agP(yR4~S)f_!u7p!xSK%
z48#FIY!Ad{K+Filt9Th0a)H<fh;4xQI}ZcH4IthL#M6N|3y7V7SPF>01MRp3#7lv=
z8Hh`PIF*}$i6L-vtHTSf$q~LHllQstPTt_fIyv5xZSz`hb>_(rj94bGu@{=`;IB5h
a!I*opuzxUEEY05+OmFrV+<d^_K>z^nI6x@?

delta 220
zcmZoTz}RqraYKLrqt50)fz9%~D>xY#O1Kyp7z7v?bS4WLY-cRneA1wob8>?v&*TZF
zLYqshCa~7u1gc2ohNzMOQXsI0pMjwah&6%u8Xp703Lst%#3?`=48&eQEDFT?co`TP
zfj9w(1Av$th~MxqFq{SA)j(Va#9=_J2gKY!{0wN%P9UBR#GOD~$<4sTkh;0m;RV;`
iNbhv!$!qL|CL8$cPG<KHo*d`zJ9)jo;N~O#HUa>>;Vy~*

diff --git a/egs/librispeech/ASR/incremental_transf/model.py b/egs/librispeech/ASR/incremental_transf/model.py
index 9f9b39769..61ecb08d1 100644
--- a/egs/librispeech/ASR/incremental_transf/model.py
+++ b/egs/librispeech/ASR/incremental_transf/model.py
@@ -244,11 +244,12 @@ class Interformer(nn.Module):
         x: torch.Tensor,
         x_lens: torch.Tensor,
     ):
-        encoder_out, x_lens, pt_layer_outputs = self.pt_encoder(x, 
-                                                             x_lens, 
-                                                             warmup=warmup, 
-                                                             get_layer_output=True
-                                            )
+        with torch.no_grad():
+            encoder_out, x_lens, pt_layer_outputs = self.pt_encoder(x, 
+                                                                 x_lens, 
+                                                                 warmup=warmup, 
+                                                                 get_layer_output=True
+                                                )
 
         inter_layer_outputs = self.inter_encoder(pt_layer_outputs)
         loss = 0