mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Merge branch 'scaled_adam_exp823' into scaled_adam_exp843
This commit is contained in:
commit
f688066517
@ -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,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user