From d1256f4c0ddd23a3af0728a1eee8f1c01af58b11 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 10 Dec 2022 14:54:39 +0900
Subject: [PATCH] from local

---
 .../.train.py.swp                             | Bin 110592 -> 110592 bytes
 .../train.py                                  |   2 +-
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
index bbbe2342221b72091652b561464e29a2a292a972..513e65aebbf039121e7a04080705067da684c4a6 100644
GIT binary patch
delta 411
zcmWO0&ntrg0LJmh?0vVrexzjAxp*(Mc37k)<))>H4B5$$ixU@nWp55n4!jc%l)`DD
zH3tqZauim}Y1aONIgXp}Q=fVc&*5<!rrR*X%dzz2RR3r+CekDMBFEVQqZDp_7_Lu>
z7eb$@e56}ybx9Qzaf1Rp9AOrNXvcS_RKqKZ@Zi8iy+eA0hXXjsVH@*Ez=nZ$OS-_W
zrMxnfCG_GgB<*1ZS@hv6DBa@@*Eqo%7BPk)Gy|N&1Fn$62BNSLz_TfxV-87-VAx!c
z{tW3CRlML9huB0K(@0?gHZ1)3+2aXiTw(_fGKl+2m3`moc(N^6-m11jg_+uF`K-QP
F`3L*&MgRZ+

delta 407
zcmWN{KPW^20LSs4>)mtjJeR?}Q6@H?QT|-nbX_(~Z*a1>Sk7X(D+48@doqyCWP0}`
z8B7LF;p&v~r;9L<7^GzI{ne*_{rVN_X0dKM6Yl76roFez6={+3ZGQIA2%L}0=wU?4
zm8`3g!dOD;hJ~kQ=^p2(;0(*iq7O;5;?L%b3PKFR!Dka2P7tDu0Bgv@!vIWN$EAa~
z3i2r>q+#I5lD4sh3=(*aNiVp=1$MBGIZUDhNxU~ow<u#5Q}B?&w<*=If(6WC#vGS=
zC@yULMx|$*Vh^iW!XoD3VF;bD@nuMFxWPU`_{bSS^*FMbN;ky9{@RZfPCsl^{{fYH
BNmc*=

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
index 23c7a5530..0543e1bf2 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -575,7 +575,7 @@ def get_decoder_model(params: AttributeDict) -> nn.Module:
 
 def get_joiner_model(params: AttributeDict) -> nn.Module:
     joiner = Joiner(
-        encoder_dim=int(params.encoder_dims.split(",")[-1]),
+        encoder_dim=params.encoder_dim if params.encoder_type == 'd2v' else int(params.encoder_dims.split(",")[-1]),
         decoder_dim=params.decoder_dim,
         joiner_dim=params.joiner_dim,
         vocab_size=params.vocab_size,