From 23e6d2e6d80bfe8dd21b666045c30d18cd9f01cd Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sat, 16 Jul 2022 06:53:44 +0800 Subject: [PATCH] Fix to the fix --- 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 de8b8681e..6b86e97d7 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"] * step / + this_weight = (group["param_cov_freshness"] * param_cov_period / (step + param_cov_period)) batch_size = p.shape[0]