Revert dropout on attention scores to 0.0.
This commit is contained in:
parent
2a50def7c6
commit
7d8e460a53
@ -263,7 +263,7 @@ class ConformerEncoderLayer(nn.Module):
|
|||||||
self.d_model = d_model
|
self.d_model = d_model
|
||||||
|
|
||||||
self.self_attn = RelPositionMultiheadAttention(
|
self.self_attn = RelPositionMultiheadAttention(
|
||||||
d_model, nhead, dropout=dropout,
|
d_model, nhead, dropout=0.0,
|
||||||
)
|
)
|
||||||
|
|
||||||
self.feed_forward1 = FeedforwardModule(d_model,
|
self.feed_forward1 = FeedforwardModule(d_model,
|
||||||
|
|||||||
Reference in New Issue
Block a user