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

---
 .../incremental_transf/.identity_train.py.swp | Bin 61440 -> 61440 bytes
 .../ASR/incremental_transf/identity_train.py  |  14 +++++---------
 2 files changed, 5 insertions(+), 9 deletions(-)

diff --git a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp
index 7d596c4123e3e61787b4ad7a814d2ae125e3f442..3f481ef80b209d33a847ca58aa1477e04d04da3d 100644
GIT binary patch
delta 254
zcmZp8z})bFSuDvQ%+puFQqO<^2m}}yQorm@R%YEORwckFvw5OGaXs&99tH+;UWlyB
zWWfp1s;hv!i9lQc#Qs3+3B(>i3{v$3sQNSzZ`wR@f-2ACmN}A8EDoflCYQ~Ts#k!5
zqSTVoqCAD%{FKxj1_qFzCYS`VGV@AOi{exBl7W&%3bqPRmJVD&K?#y9f&)<iHWH}I
X6J$zSeo<~}kp@^Nh}gVou3r!UZtFz-

delta 110
zcmZp8z})bFSuDvQ%+puFQqO<^2m}}y!oTcJ{>HLVtV)1UVe>?R;(FeH+zbq}cp$P0
zlLaS4tFGf^V3-EPwLqK##J)hR3dDbbs;&d^Mj&3jdEx|Bp2_y}{%+np*Ea|NCAl3c

diff --git a/egs/librispeech/ASR/incremental_transf/identity_train.py b/egs/librispeech/ASR/incremental_transf/identity_train.py
index 8b9ba6747..e690dc48b 100755
--- a/egs/librispeech/ASR/incremental_transf/identity_train.py
+++ b/egs/librispeech/ASR/incremental_transf/identity_train.py
@@ -499,16 +499,12 @@ def get_transducer_model(params: AttributeDict) -> nn.Module:
 
 def get_interformer_model(pt_encoder, params: AttributeDict) -> nn.Module:
     encoder = get_encoder_model(params)
-
-    model = Transducer(
-        encoder=encoder,
-        decoder=decoder,
-        joiner=joiner,
-        encoder_dim=params.encoder_dim,
-        decoder_dim=params.decoder_dim,
-        joiner_dim=params.joiner_dim,
-        vocab_size=params.vocab_size,
+    
+    model = Interformer(
+        pt_encoder = pt_encoder,
+        inter_encoder = encoder,
     )
+
     return model