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

---
 .../.train.py.swp                             | Bin 81920 -> 86016 bytes
 .../train.py                                  |  54 ++++++++++--------
 2 files changed, 30 insertions(+), 24 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 31241ef2b343323eca3bc2e81230b7f7f0bac6ab..a772af123903de5c85176c4a202edf431c0caa2e 100644
GIT binary patch
delta 1672
zcmb8vPe>F|9KiAS%0KI$)iqQ5qqZto(z+@b+94qd!od6!NkV0RTr1lOj1a<FA?PF$
zPB3&TA(2+gY7Uheb;wHxDI(}lT{=V^EGq~zsPCH{rwPZR2cP%eyg$F+%)om+J@GwP
z<FgKyH`mtrHwMcE#V5q3)Jumh-_Dv#mt9(hFQ;U=8mCmn8gblerPVI8+NDOL@0&E%
zExYIK_s4k;Gu*m_ijZAKhTo7|E0zk8k3_s>>NOr?7?*JY?br{sHGpCiz=H#)!+yOW
zAoH}k!eVV;bh(y$Gf9h%rs@-=@~l3!UWSdmMMFO#&(ejMh00AXv&`}Q-5UQ&JB?R(
zf<g4+JkFsTAvB>9tC0;Ko~H>hfhaEG6gm(>E0WU0G4a_e#2ZXv1j86aJ*rTIeB|Oo
zDnaoGL+C><!f3%Z1hEpC_?RNZOHATUiXr2!Bx_%y8B%C7(ag1ctDjP0)~a_T<Tk|G
zY;#Ap`fc8}*cxl4x!0^zm{`C+$NZ`+oLD&GT%->KWY$q~5X-D+%s8t0&rAfcx0*fc
zgB`;vU5?%mkiHtSsYQ-Sdym@~OPy=&K$R!0)r)O%?y@teXs#2BL~#uN&K#M~?2Kiv
zv^&-x{p-(kc^9|+&P0B{{U)o++}^~ISADHBh1!|9&n1J)P;9F?Vb9u~f8@xiE*H+c
zLIo=k3%6S%YChJz`?E|+U*rIXpe<R5J)A3bam?R$?>O{R7{@hqB8<b>fek1?5)3YY
z8yJCxGiXOWW;yF$FpU=&$942$msdp0%V9eY6<C1;%y8f*@epGeKsP$D8%=0LHC7=J
zpZI_ec#o$rFo0f!(T)(*hwenI7)f!Pll7W>>B)=z7VAW*-$)T=O{H!QZMyKJzQ!Zd
z^;-eiV)s1iv^8@%_v-h9a+RK6D*Ns}2+A!|i;Sh{4P`PuA#~zUX}D9{T$HQT7OvDc
H24&ALDF*Yp

delta 494
zcmXBPKWGzS7{~GFi7mGIdx^OQTs%|6L69U)ItfLj^rA#coD$SHI22qOG(z<@aVmJN
zx2A%oIt0QEmpIrAg=`&4K@<c9M-hWVDe2(%=np>c`@HYV`_E}lJI=KEpio^bTURd^
zgcu^D*hVALGbbr+xoIOednTMG>Q)$KP6kr+E5@M)j|}Z6URzU%aCad>!wD@vZTY%z
z@sx-O!4H<c;VXvdU=t3OaSMwGk2sAa?l)K5-a8XnZ_CyX-nkhyy=P1M*sslZj$eF<
ziUhHfD~ohtey2nZ@ElL^1a;g%5ofWR6#0w++IWdo)NlzG5XZlS$WQE{i*2l<iAF;1
zicOKiEdItt_OXL)yu~X##2sA6HJnEh6CVBxLwv#jeFQOIr$*77RVrPv2!En1g(VSp
y{Bh15=U3eC`J{Kx*1A`<HS@;S3Nu-+QPy?$sB5^Z`8ChEtxoW#tT*+2L5=@Xr(d@K

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 dbfc2863c..7048859f9 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -105,17 +105,23 @@ def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None:
 
 def add_rep_arguments(parser: argparse.ArgumentParser):
     parser.add_argument(
-        "--decode-interval",
+        "--wandb",
+        type=bool,
+        default=False,
+        help="Use wandb for MLOps",
+    )
+    parser.add_argument(
+        "--accum-grads",
         type=int,
-        default=200,
-        help="decode interval",
+        default=1,
+        help="accum-grad num.",
     )
 
     parser.add_argument(
-        "--encoder-dim",
-        type=int,
-        default=768,
-        help="encoder embedding dimension",
+        "--multi-optim",
+        type=bool,
+        default=False,
+        help="use sperate optimizer (enc / dec)",
     )
     
     parser.add_argument(
@@ -132,41 +138,41 @@ def add_rep_arguments(parser: argparse.ArgumentParser):
         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)",
-    )
-    parser.add_argument(
-        "--accum-grads",
-        type=int,
-        default=1,
-        help="accum-grad num.",
-    )
     parser.add_argument(
         "--encoder-type",
         type=str,
         default='d2v',
         help="Type of encoder (e.g. conformer, w2v, d2v...",
     )
+    
     parser.add_argument(
-        "--additional-block",
-        type=bool,
-        default=False,
+        "--encoder-dim",
+        type=int,
+        default=768,
+        help="encoder embedding dimension",
     )
+    
     parser.add_argument(
         "--freeze-finetune-updates",
         type=int,
         default=0
     )
+
     parser.add_argument(
-        "--wandb",
+        "--additional-block",
         type=bool,
         default=False,
-        help="Use wandb for MLOps",
     )
 
+    parser.add_argument(
+        "--decode-interval",
+        type=int,
+        default=200,
+        help="decode interval",
+    )
+
+        
+    
 
 
 def add_model_arguments(parser: argparse.ArgumentParser):