From b7ecb3b6208ea0eadc116e1f341dd3bfd09f0f8a Mon Sep 17 00:00:00 2001 From: pkufool Date: Thu, 7 Apr 2022 13:03:10 +0800 Subject: [PATCH] Minor fix --- egs/librispeech/ASR/pruned_transducer_stateless/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless/train.py b/egs/librispeech/ASR/pruned_transducer_stateless/train.py index 96893e4fb..e743106ec 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless/train.py @@ -650,7 +650,7 @@ def train_one_epoch( optimizer.step() if old_parameters is not None: - deltas = optim_step_and_measure_param_change(old_parameters) + deltas = optim_step_and_measure_param_change(model, old_parameters) tb_writer.add_scalars( "train/relative_param_change_per_minibatch", deltas,