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

---
 .../.train.py.swp                             | Bin 73728 -> 81920 bytes
 .../train.py                                  |  38 ++++++++++++++----
 2 files changed, 30 insertions(+), 8 deletions(-)

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 d13a7bd4153fe1605c8d935031d72fa8b370c994..f8f13fc789f03fa7649ab415e151ec6e18d5bced 100644
GIT binary patch
delta 838
zcmb`_&1(}u7zXg!BwA~mY`$wtiIen@R=Ra_@be%o>LG_JmC&QG$qvopb~kP|YAk5}
z07Z1vToe>6LZOwILk<<KD2f;H;6=UkR*+nJ@SqocCQUAaHwS(UGw;qjlZVtce6`xH
zcwU{)&eCI3D!~X6@@wziLjU(hye-RNavggfQNXD3P;e+4*;x>&J1_mU4tLklf7C<W
z{lR7?cuT%ReSUGVDlWn0)n<q`rITqdAu%AZiPBg20vqrY?m-Q5a0ap<TJ%8xw2H|a
z%Q1+;u#aN!$IoJY%qrMAb9iz!TE8tz5$}=1NBPAk`#sF!$(7@>&P1IWmdl(=n%Vh=
ze|ny(PY+1D`<32)b7t&KJ3AL3<ROTM4`LGIolM9o+=L}K55o|FF#N>#-{3>{>;ugA
zTeCqa8>m0>Nc&o66Vf?}X*wb6enQeHi}t-V+i39_uE7<!4C0E@kbx0+i!(1_8E!xo
z@}R;2n1FFex8{P<YrlsQzP%DpR?3W)OU%(+M(vVoEE-ptL&uRAJxGzu#7-D%t>|dp
zw#`W*LKP#WD2o--HI&XNJ~KATf2>7M<%%Ll!!<OMnoM&n!&;!C`xK?QqEV*Ew}LB+
zcE!}GWxLd3OsB3*^NbdXnzg`m9$Fvrs2QGkl^iHB?Sg`LD5k@=UmTkLAHBV5r+Iut
p<ozDmZGp(U_wU}4n^O3<zp?pIs-H+mdqViJorfS6JR~(q{s7FS!A1Z8

delta 283
zcmZo@U~M?SB9>$j=IN_osb|0d1Of~U;apRa7rxvm_J*HdfT12J#src8g3`%?0^*#4
z5Do+X=0bsf{)rD9CMmG+x&Q@DK<a?{OE(KDY-Ti7VP;?u0^+Yg;SE5%3W!$%@dO|)
z0^$T925I@h2sIE$DNMJMW3*&a0MikZpRLiEykR}tG<HU=>67^wHK#|(GnQ;lRi6Kb
z_Zcq(gB>5lLjK8u3xqY(`574efLITRWq|lUP<9s(Zvo=XKs*bGYk)XwyRQJFCm&YB
Omree(N_6tJ^<DsISv<x7

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 c72936741..db6fa3144 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -105,18 +105,40 @@ def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None:
 
 def add_rep_arguments(parser: argparse.ArgumentParser):
     parser.add_argument(
-        "--num-encoder-layers",
-        type=str,
-        default="2,4,3,2,4",
-        help="Number of zipformer encoder layers, comma separated.",
+        "--decode-interval",
+        type=int,
+        default=200,
+        help="decode interval",
     )
 
     parser.add_argument(
-        "--feedforward-dims",
-        type=str,
-        default="1024,1024,2048,2048,1024",
-        help="Feedforward dimension of the zipformer encoder layers, comma separated.",
+        "--encoder-dim",
+        type=int,
+        default=768,
+        help="encoder embedding dimension",
     )
+    
+    parser.add_argument(
+        "--peak-enc-lr",
+        type=float,
+        default=0.0001,
+        help="The initial learning rate.  This value should not need to be changed.",
+    )
+
+    parser.add_argument(
+        "--peak-dec-lr",
+        type=float,
+        default=0.001,
+        help="The initial learning rate.  This value should not need to be changed.",
+    )
+    parser.add_argument(
+        "--multi-optim",
+        type=bool,
+        default=False,
+        help="use sperate optimizer (enc / dec)",
+    )
+
+
 
 def add_model_arguments(parser: argparse.ArgumentParser):
     parser.add_argument(