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

---
 .../.train.py.swp                             | Bin 81920 -> 81920 bytes
 .../train.py                                  |  22 ++++++++++++++++++
 2 files changed, 22 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 3f4504aa59b5b5ad4f448d56c10d91cffa38dcac..86325ba07c2e95397e7705d82bc3abe613fdc7b3 100644
GIT binary patch
delta 402
zcmYMuJxc>I7zgkKJW#GLSEzLHB`Q)&dq?R~s|Y#@id4k?F3FYR^-|8eYMl<8oE#!_
za<EfzC`gBbB93kjegnUOAg<yQ75u|5B;o&r#IY8~+O$(X+}<TMtBfE9LQ=n?_jl9l
z1}>i0jd9(;qjXZ}2vrlmg<>y5@Sc-MA@q?%=ncGpAsB#55CIByKpISecNl*KPhbH0
z-~uqP4>rLXsDV{bO^oowsh%T{0#d6A<E|Zf;YL~F+qs+@aDK*3x9RgR^0^tcnH}<D
zxwSN-sdhnLkCzu5A-ob&XO%_Mv>9uLP2t;~>3D)3slNOiW{U^iHYdU*+^2%^fUIz<
zVG$~PR|G8{l*mb?QzER=u`HQ+$ZC1m=B3@A*Z(HWu-le5kF)Bvluc>WG8=)-h#$4o
VK<{f36aD}G=clSWu37jw^9`ywY5M>G

delta 98
zcmZo@U~On%6-_b-^Ym4))H7fJ0s#hw3EWeXCu|h87GM-@4iwlPD8RUxpLY{41A`JD
sM1kmZK?O#CP8A?m8HklYN>=brx6)zcW64O(VVL}JmDqMmJ;vjL00tfvh5!Hn

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 e1f1d1f42..12034bb33 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -138,6 +138,28 @@ def add_rep_arguments(parser: argparse.ArgumentParser):
         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,
+    )
+    parser.add_argument(
+        "--freeze-finetune-updates",
+        type=int,
+        default=0
+    )