from local

2025-12-11 06:55:27 +00:00 · 2022-12-10 13:51:32 +09:00 · 2022-12-10 13:51:32 +09:00 · 30c8dbce02
commit 30c8dbce02
parent 2c449264b5
2 changed files with 6 additions and 0 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
--- 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...')