Increase pos_emb_skip_rate from 0.05 to 0.075.

This commit is contained in:
Daniel Povey 2022-11-15 11:50:14 +08:00
parent f76075fd1a
commit 6668814940

View File

@ -1030,7 +1030,7 @@ class RelPositionMultiheadAttentionWeights(nn.Module):
query_head_dim: int,
pos_head_dim: int,
dropout: float = 0.0,
pos_emb_skip_rate: FloatLike = 0.05,
pos_emb_skip_rate: FloatLike = 0.075,
) -> None:
super().__init__()
self.embed_dim = embed_dim