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 081ad5e89..d122c599d 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 70e1f37a8..fd25337fd 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -1185,16 +1185,16 @@ def run(rank, world_size, args, wb=None): logging.info(f"enc name len = {len(enc_names)}") logging.info(f"enc param len = {len(enc_param)}") - logging.info(model.parameters()) + #logging.info(model.parameters()) optimizer_enc = ScaledAdam( - enc_param, + *enc_param, lr=params.peak_enc_lr, clipping_scale=None, parameters_names=enc_names, ) optimizer_dec = ScaledAdam( - dec_param, + *dec_param, lr=params.peak_dec_lr, clipping_scale=None, parameters_names=dec_names,