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

---
 .../.train.py.swp                             | Bin 86016 -> 86016 bytes
 .../train.py                                  |   4 ++--
 2 files changed, 2 insertions(+), 2 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 4b2c63e0da7c263f408be32100dc196089224455..76d8fe722beb77ba40cd5cd8719237811d896545 100644
GIT binary patch
delta 371
zcmXBQK`4W97{~GFzt^k0HXQ6-iHsfOKvPm`$w~1xQc`P9@7uI?*hzA;xHxfm$!YaE
zh@IY~{F4K24pNRg?S#@s4(`6ouRitkTz)-IB`_<2X}jt5#cXnRGA*P}<TiTjS<Nf^
zYuNPI<gdGu_iB8=5HZk*h}7|b3!GpNE;f+&e@06FSs3@XMzx>N@%>CnXHx2EaJOKI
z<huV$X4+-sqgUhw8Ue~E!ow1#Fa#4}k4O!dIKnnuIQWf;eB%?Z&^X5-R^h~?Bnu1^
z=tl&tsK_JkaDys#u!%J+Ba3O+Fwo?Hcf8>lA*wh)5iWANHLuE+-<a#v1xI-u?LVB>
HSJc=azb{1c

delta 384
zcmXBQKS%-r6vy#*)hXIt*&-HEcREB%*iaNWwD!-VplA`JK<SW(hK2|df~KY}kT!*u
z{uehK<ls`!6fF%6O%73!Bu#C7tsi{g@s=O&<%?>*sD_u~^W)RT=s;XZNaRX8Sgw9t
zhikoRmQC6X7!{?zr&+|0YLmzdZgGl3>>`6jq@B8HJCO_BIl0uFwWwDJO(`8aiI|s&
zDfivmD?yP|>wnX5qmjI}i#(u=BdjBbERq<4i2&R-k#n5E!6q`W@S}@-qlPDxag2S;
z!P3Q+35G#L;6q&#xkCjvC}9gW7BGt$jGzmCd~v`#-tY()CF~)G3{u|bn6eu5iP4B-
TMmxQYNhR6x_m9G}rS$y*3N}Y5

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 08a10a9ff..378074344 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -106,7 +106,7 @@ def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None:
 def add_rep_arguments(parser: argparse.ArgumentParser):
     parser.add_argument(
         "--wandb",
-        type=bool,
+        type=str2bool,
         default=False,
         help="Use wandb for MLOps",
     )
@@ -119,7 +119,7 @@ def add_rep_arguments(parser: argparse.ArgumentParser):
 
     parser.add_argument(
         "--multi-optim",
-        type=bool,
+        type=str2bool,
         default=False,
         help="use sperate optimizer (enc / dec)",
     )