From fde2035d3657002debafd44703593a6632afbfce Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 10 Dec 2022 14:54:11 +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 f3b36ecfbebcbfa3cd56110bec03f8aaae6c63fa..bbbe2342221b72091652b561464e29a2a292a972 100644
GIT binary patch
delta 359
zcmWN@KPW^20LSs)yWYKf$4ink>P<}al*Pc6?YeF?-hi{X7@Wm&R|XcO_v!{>VllmY
zLI#5jGWe5I%5_}|!zm0Flkc}a^{uq5O3O-3dsCU5J2vczx}{QoF!LIwzV6~@F{$V$
z@BQ+GBlY3CQ@X(wBGeFK3nk<+4hv5m(g_0i$il>VQrg81a_GW`Exn_GTkK&Q%b2xQ
z)&P^jXF_^G4F{M*9%=kqQXLyu#Ud6k0uK)U;?f(gafD5*V+||FV-kaK@MB7CvsZd%
tju9b1(F`9hV%2oku%p9zGf}<VH+rh=QNy*|LUF0!`t!}9=(Le}{0Cu6M?(Mr

delta 344
zcmWO1Jxf9X0EOXm#rvft4N5g6QbS9GQ`BN@6%jN=t@;BraE<E^Xt-dMEs{$u4T?n6
zQV}`zp%64BLKYE0O${Q{d*Fe0J%x@{=vc9Ia(#I<F*ltQ8<O1kZf2(z`+hV(15&x@
zHj7?vKw82$`eEr8Z|LF`*EoWYEM{TjBP2C&i*3x{%atB*4-X3%#Rwu0nvQgXW9&Kd
zv_UK)jzM&7=@}*X*uo?%Tw2m8DmX+2X{0cPVf0Mt1y{JhIZogsk4>a7fiW{Gg^2ec
fuhBp`SZn<R_UBg&r=EXm+h>0V#+^}LYgb$U)%rlP

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 ad95d67bb..23c7a5530 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -592,7 +592,7 @@ def get_transducer_model(params: AttributeDict) -> nn.Module:
         encoder=encoder,
         decoder=decoder,
         joiner=joiner,
-        encoder_dim=768, #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,