From 15b91c12d65097a89aac5af2e701c310cb42a7f1 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Fri, 14 Oct 2022 15:14:06 +0800 Subject: [PATCH] Reduce stats period from 10 to 4. --- egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py b/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py index 7dede581a..b0761f045 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py @@ -258,7 +258,7 @@ class ActivationBalancer(torch.nn.Module): max_var_per_eig: float = 0.0, beta: float = 0.75, prob: float = 0.25, - stats_period: int = 10, + stats_period: int = 4, ): super(ActivationBalancer, self).__init__() self.num_channels = num_channels