From e63b77a6ad4ce0b18e8fcd2f82a778352f011fa8 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Mon, 26 Dec 2022 13:57:41 +0900 Subject: [PATCH] from local --- .../.train.py.swp | Bin 94208 -> 94208 bytes .../train.py | 7 +++++++ 2 files changed, 7 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 6c99a21d1ad26ea4852687e50842e2632aa3263a..80da68d907adc82af32bed0e4a79c0dae3f2eedd 100644 GIT binary patch delta 419 zcmXBQJxD@v7{%fDpIy9__Hp3`eGH)l2^CgTL_>p+WG#Iy!o}1Sius!=s>wkuRiA4{qJUq4h(NdN!< delta 341 zcmXBNtxE%O0LJm>=iKgecREgHaK#G-i|ook!C@5{BB(*|1uZi=7F^oF{)7W3f|!l6 zaJXO-1dFmX5xrnB7&MujU-5wtJmG;SOwwVJ&hHmDmn+lrb48Jyv^slvP`}C-JX3B3 zUL1I)x7fVRNPX*nQ2rZ|!MOB>Q?wD{5LK*U0Rd7-#-wvJuz>(B-d(AS6Vz~o3QCy3 zmm}RdYU+k^h8Fg)gh@EaqMPL$C8RKf?@{Rq*QjC#+wk!`BK6QgggVwSg94`TXiE=h k*c@z8eDtl38NSa%ub*~w{NlzRzPYfXm7iKX*fl2i3p8pvH2?qr 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 cbc3430ab..ccf4b5a84 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -1521,6 +1521,13 @@ def run_adapter(rank, world_size, args, wb=None): if 'adapters' in n: adapter_names.append(n) adapter_param.append(p) + + optimizer_adapter = ScaledAdam( + adapter_, + lr=params.peak_enc_lr, + clipping_scale=None, + parameters_names=[enc_names], + ) if params.multi_optim: logging.info("Using seperate optimizers over encoder, decoder ...")