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

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

diff --git a/egs/librispeech/ASR/incremental_transf/.model.py.swp b/egs/librispeech/ASR/incremental_transf/.model.py.swp
index 93db73defa70a915f53fc79762cc56dc517f7f5a..3c123591e554f0f91ba9f4afd2c2afe04688d7ef 100644
GIT binary patch
delta 135
zcmZoTz}RqrQ9Q{Y%+puFQqO<^2m}}y7Jb~ET+hI}QT(kv>tikkhWyE#4%Uv1ybKJ5
zK&%7Ayg>Yphk@Y?5YGqVav=5xVjUo62jW}Y3=BJfcoh)O0^*rKoWHr%;WgLfBv-S|
U`@H`%PCn-=&ZxP0udlNZ06-xieE<Le

delta 149
zcmZoTz}RqrQ9Q{Y%+puFQqO<^2m}}y_&)4Tp7oDsqxf5U*56zV4DFLS9jqN^@-i@V
z0<jAas{%0_5I^BzVAu}C{Xm=!#4bQA55%9i85k}A@gX4I1jHMGxP5c0!)va|1+F5K
j6+Nvc*ZK-f-si?Ud4m`0<akfE&F8$YGj87N>nsESg6$^B

diff --git a/egs/librispeech/ASR/incremental_transf/model.py b/egs/librispeech/ASR/incremental_transf/model.py
index 29f520300..ccf97aff4 100644
--- a/egs/librispeech/ASR/incremental_transf/model.py
+++ b/egs/librispeech/ASR/incremental_transf/model.py
@@ -255,7 +255,7 @@ class Interformer(nn.Module):
         
         with torch.cuda.amp.autocast(enabled=False):
             for inter_output, pt_output in zip(inter_layer_outputs, pt_layer_outputs):
-                mse_loss = self.mse(
+                mse_loss = self.mse(inter_output, pt_output)
             '''
             simple_loss, (px_grad, py_grad) = k2.rnnt_loss_smoothed(
                 lm=lm.float(),