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