diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py index f02813c78..26e977692 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py @@ -1055,7 +1055,7 @@ class RelPositionMultiheadAttentionWeights(nn.Module): initial_scale=query_head_dim**-0.25) self.whiten_keys = Whiten(num_groups=num_heads, - whitening_limit=_whitening_schedule(2.0), + whitening_limit=_whitening_schedule(3.0), prob=(0.025, 0.25), grad_scale=0.025)