From c4d233e809267bc27768d50bec0404079169b38e Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 10 Dec 2022 13:24:34 +0900
Subject: [PATCH] from local

---
 .../.train.py.swp                             | Bin 65536 -> 69632 bytes
 .../train.py                                  |   7 +++++++
 2 files changed, 7 insertions(+)

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 0accbf9ea19692557659be70de4b20efaf4e3d25..06e6e86bd1df3e51e0928b826bdfe1c1621ab391 100644
GIT binary patch
delta 1219
zcmXxiTS!x30KoB2H!s_?&C7^Fa;f#8c_}2*9*PRXYLq2CSWxPgjBsMpmlQS;10m|+
z%ZjuzLuIhEwJftz>?YbIhp8EKdFVpg$Sew>vi~AK_<iRb&UX&y@^?6tjuxf9IAedF
z(U6^%As9L#YE@@Se!BI3ZID(fbg4l)xmLF{iWC1v_5W?HCtKH{y<t!~B2`O-Fe7lW
zPDZt)*k?JKL?2u@hocCrihu@lkwQ%15iVjQ)KEf!=?EbP(T!6`$B0IVUfhHS7huIf
zY{I-+h;M3P7jx`P;SM@b4<oiC7paIvF#IZRhHG#^q8b(yz<@X5LfnEASJ8?qI1LMu
zk%UC3!|l9Xm=JF<f@YjWBWh5BeB>Ym3!y^%!Fzbojms!R0rnvW3DDtJh!8$J#W37B
zj}x#V#m@9y$pZAD2e)wv^|0YER^ydIh*1n<2-nes2IM0ZUxN9PaG@P-IEgBhAQOwj
z!0;GbeHyv4GgcO*XUW+#y?j=smg${g%dI8lCi5|axvbc1GFhyp29u@CY_(a+t<v2d
zqsUsnLDo3qrMFWf+njOAz|G`<Q!o2F)Sj3w{~mjhNr+uRd>0-o(A#2;3tAyQz>8i;
ztiu|lAR0fY?j7#q0Fse}M5ytR>c-KD-SAV~OFYLIdLt++kYWuEVHsi&jd}8$!XVmE
zhEg1X5t&FpB*v(21YUU1ga%Y%GZrXr0z-I+2N*z?in1Em*@11?ieyAXz)ybn5J<Hf
z&8R~K3b6|DSb-3HroM6X<2Wi|Lm{>x19A9Dbu*Z@Gx^Yic1Y|*EM}<gHC|v6eQ1M(
t1G)H3akH4lCye4QoTz{iT1@eKy?BDhaG?>QbxZ{cw{LHdC$H5N@fV;E&>;W-

delta 608
zcmYk(O(??w9LDkA?9Vo4wzlP^v^hu;aS%%)IdG62mJ26JoaCT(vXvc{ll6BHrO8Y3
z(lB{XP0_r~!AWGJxQK(exbT&Wr#`>abNJQc4=aAZV(+f(Xl--4Dk_DTMd}UXJ-h4n
z*Bp&W)tbvS0kz7U8|^bs<b+br4X5JIE~MC+_SX)mmT;Nx!z6Nx7$OLw1VwN{hnH-T
z8*IQ02lA1JZ==W^&M=E+JZ6bp;S_O1FoHpNV1gdHEb+-(rbq$@m_#?a(1|9LA{S|h
z8$^x}!y?8ojDA$(LoafUL+oH1TbRQz8sI`L9MI`~BA+@w#yXa<gb=*wMLP;$gB4#H
zA`dvjCI--tKD46-#jxNxT_lAhPOyq81knhCkCReFu5pPAL@|jVhERuht;iFSxW_Tp
uFpXX`K|z9XW7x+YW-tmbeur<t|Adm3ZI4#7VpP8uY-(~oGir^zx_<yrp=1jH

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 8dac60025..5681003e5 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -124,6 +124,13 @@ def add_model_arguments(parser: argparse.ArgumentParser):
         default="8,8,8,8,8",
         help="Number of attention heads in the zipformer encoder layers.",
     )
+    
+    parser.add_argument(
+        "--encoder-dim",
+        type=int,
+        default=768,
+        help="Encoder embedding dimension",
+    )
 
     parser.add_argument(
         "--encoder-dims",