From a8333e49fcdf435db5fabd1e1750350748c4edb4 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 10 Jan 2023 01:49:34 +0900 Subject: [PATCH] from local --- .../.train_adapter.py.swp | Bin 77824 -> 81920 bytes .../train_adapter.py | 8 +++++++- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp index c690be68878c392236e2e9675ce531215d37837f..e031dda8e6746492a7a9b1166b45678bf329b23b 100644 GIT binary patch delta 1013 zcmXxiYe-XZ7{~GF!uwyk+kcO>|m1Q+&UFY@3(G87C( ziCjYv-Dq3MR)svgUm)@dFL4(a(S@2GB~?*7FX%$OWzB}3nN+4dU!_mn)#yGqZ%H#u@#XN+u|5_48;Voe23>4 z#~F0MjUC9uG>M$TB*rj+b2tbalCT7GBytktxCtLBQGp_4Vg=^tc;8aCljuS#4xkWe QF(E~FBKq%?pBfbT4XzldSpWb4 delta 523 zcmXBRPbh-{7{~GFwRba&jnaG5Yvn-EoR%z>GC3%P*-pd(Oc(aP+@q#PtV;y01zymjG zU`C->B!?spk-#!0Fo;gHqYZT^#-~N(84tL}DRwMjkp+fX_|c0l_+Uf9Oc1^O!>j0r*gloQVq~iwxpeK}e^SLWK$MJm(JQ_G(qWOxIJ@ur) Utvp(d=}_9JTB=TYCoisl0WJMm`v3p{ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py index 02bcb47eb..e260add47 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py @@ -161,7 +161,6 @@ def add_adapter_arguments(parser: argparse.ArgumentParser): ) - def add_rep_arguments(parser: argparse.ArgumentParser): parser.add_argument( "--wandb", @@ -169,6 +168,13 @@ def add_rep_arguments(parser: argparse.ArgumentParser): default=True, help="Use wandb for MLOps", ) + parser.add_argument( + "--hpo", + type=str2bool, + default=False, + help="Use small db for HPO", + ) + parser.add_argument( "--accum-grads", type=int,