Merge branch 'scaled_adam_exp891' into scaled_adam_exp896

This commit is contained in:
Daniel Povey 2023-01-15 12:52:41 +08:00
commit fb30d11693

View File

@ -489,7 +489,7 @@ class ZipformerEncoderLayer(nn.Module):
dropout)
self.nonlin_attention = NonlinAttention(embed_dim,
hidden_channels=embed_dim // 4)
hidden_channels=embed_dim // 2)
self.small_conv_module = SmallConvolutionModule(embed_dim)