From 68c59356911d7caedc0fef533c3f38ca0dea8eb8 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Fri, 15 Jul 2022 08:33:10 +0800 Subject: [PATCH] Fix bug re param_cov freshness, properly. --- egs/librispeech/ASR/pruned_transducer_stateless7/optim.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py b/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py index 84a2d7811..7cd69feb4 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py @@ -469,7 +469,7 @@ param_rms_smooth1: Smoothing proportion for parameter matrix, if assumed rank of param_cov_period = group["param_cov_period"] step = state["step"] - this_weight = (group["param_cov_freshness"] * param_cov_period / + this_weight = (group["param_cov_freshness"] * step / (step + param_cov_period)) batch_size = p.shape[0]