From 30c8dbce02a2e7635910677b6f2114680b3ae978 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Sat, 10 Dec 2022 13:51:32 +0900 Subject: [PATCH] from local --- .../.train.py.swp | Bin 90112 -> 90112 bytes .../train.py | 6 ++++++ 2 files changed, 6 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 d32bf2066d5ac1772f603a6e426eb851342c8f10..94abd585e7835b904758585901a3d5ed5b6863ef 100644 GIT binary patch delta 213 zcmZoTz}j$tRW!*U%+puFQqO<^2m}}yYz3wy_iPmXEyx(USy1S6E^jF_1H)4mh$K*` zft%6q8#@ETZ6MwX#1nux1&G6dI24H8ftVABZ?Z8ktOVi|AeIMWIUwG}3RDNg%0T=9 zXykh!e#){rtFfAS^V-%0OiU@Mll?n<*i%xI<8z91CM$M|vKAy3CFV{JoFy(44-{6g kRR~T_%t=jgOi9eunCzb-2N+jNF@jI<{N~0MQCI82|tP delta 179 zcmZoTz}j$tRW!*U%+puFQqO<^2m}}y3Py8 uo(;qkf!K$2b5>(D^Ym$vj0Xf1kU(lqaq9HMI7ac!EUk@9n|(VrUk3ofAtrSI 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 0ceaf0efa..de55f139c 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -1153,6 +1153,12 @@ def run(rank, world_size, args, wb=None): clipping_scale=None, parameters_names=enc_names, ) + optimizer_dec = ScaledAdam( + dec_param, + lr=param.peak_dec_lr, + clipping_scale=None, + parameters_names=dec_names, + ) else: logging.info('start wandb sweep optimization...')