From f41acc98effdd40b6e98593f94b35614ef51b1b7 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Mon, 9 Jan 2023 19:53:01 +0900 Subject: [PATCH] from local --- .../ASR/incremental_transf/.model.py.swp | Bin 24576 -> 24576 bytes .../ASR/incremental_transf/model.py | 1 + 2 files changed, 1 insertion(+) diff --git a/egs/librispeech/ASR/incremental_transf/.model.py.swp b/egs/librispeech/ASR/incremental_transf/.model.py.swp index 3c123591e554f0f91ba9f4afd2c2afe04688d7ef..4783f02d6e9d54f9e0867e11defe1f28625b8779 100644 GIT binary patch delta 187 zcmZoTz}RqrQ8dXQ%+puFQqO<^2m}}yR(;%^ykethv;d>*=0btV`n>zN7#RGxA(FC_ z1s%*C6L=XI0)Utsh~MxqFq{SA)j(Va#9=_J2gKY!{0yjTClF5u;!Ys01mX%H_S@X) zu#1aZ0SbyzOC~3}noq9v6_~uwjd$_}FV@NNo@|@Xc^_sJ$jL7*R?xOp$SqEd2eCId I_%;Xu02D?oivR!s delta 166 zcmZoTz}RqrQ8dXQ%+puFQqO<^2m}}y7Jb~ET)$B?T7XezbD_Xwecs1h3=H|)5J{QI zf)3`6g}e+5IzY?|#P4_*7|sCkd>}3dVt*jk0b+I_z6I2^1Bh1v@hl*o3B(|iR&DNd v*u^zj!A))QTvztVNv>v-V|)cB?{njwyupiga=a(o=C$6388=V#%@+ay_DCol diff --git a/egs/librispeech/ASR/incremental_transf/model.py b/egs/librispeech/ASR/incremental_transf/model.py index ccf97aff4..22a2e413a 100644 --- a/egs/librispeech/ASR/incremental_transf/model.py +++ b/egs/librispeech/ASR/incremental_transf/model.py @@ -256,6 +256,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(inter_output, pt_output) + loss += mse_loss ''' simple_loss, (px_grad, py_grad) = k2.rnnt_loss_smoothed( lm=lm.float(),