diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py index c00f04e31..177aa3c3b 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py @@ -263,7 +263,7 @@ class ConformerEncoderLayer(nn.Module): self.d_model = d_model self.self_attn = RelPositionMultiheadAttention( - d_model, nhead, dropout=dropout, + d_model, nhead, dropout=0.0, ) self.feed_forward1 = FeedforwardModule(d_model,