diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py index 77b786a91..7d785a369 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py @@ -198,7 +198,10 @@ class ConformerEncoderLayer(nn.Module): # try to ensure the output is close to zero-mean (or at least, zero-median). self.balancer = ActivationBalancer( - d_model, channel_dim=-1, min_positive=0.45, max_positive=0.55, max_abs=6.0 + d_model, channel_dim=-1, + min_positive=0.45, max_positive=0.55, + max_abs=6.0, + max_var_per_eig=0.1, ) self.dropout = nn.Dropout(dropout)