Merge branch 'scaled_adam_exp823' into scaled_adam_exp843

This commit is contained in:
Daniel Povey 2023-01-04 17:02:37 +08:00
commit f688066517

View File

@ -439,11 +439,11 @@ class ZipformerEncoderLayer(nn.Module):
value_head_dim)
self.feed_forward1 = FeedforwardModule(embed_dim,
feedforward_dim,
(feedforward_dim * 3) // 4,
dropout)
self.feed_forward2 = FeedforwardModule(embed_dim,
feedforward_dim,
(feedforward_dim * 5) // 4,
dropout)
self.nonlin_attention = NonlinAttention(embed_dim,