diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py index 0b2ab23d5..35daf4f39 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py @@ -1031,7 +1031,7 @@ class RelPositionMultiheadAttentionWeights(nn.Module): pos_head_dim: int, dropout: float = 0.0, pos_emb_skip: FloatLike = ScheduledFloat((0.0, 0.5), - (4000.0, 0.05)) + (4000.0, 0.075)) ) -> None: super().__init__() self.embed_dim = embed_dim