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 826e1ebe2..6b5f3df84 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp differ 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 557ee432f..d6e15897e 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -1184,13 +1184,13 @@ def run(rank, world_size, args, wb=None): dec_param.append(p) optimizer_enc = ScaledAdam( - *enc_param, + params=*enc_param, lr=params.peak_enc_lr, clipping_scale=None, parameters_names=enc_names, ) optimizer_dec = ScaledAdam( - *dec_param, + params=*dec_param, lr=params.peak_dec_lr, clipping_scale=None, parameters_names=dec_names,