Yuchan commited on
Commit
411d64d
·
verified ·
1 Parent(s): 133d2fa

Update AlphaS2S.py

Browse files
Files changed (1) hide show
  1. AlphaS2S.py +1 -1
AlphaS2S.py CHANGED
@@ -24,7 +24,7 @@ class EncoderBlock(layers.Layer):
24
  def __init__(self, d_model, num_heads, dff, dropout=0.1):
25
  super().__init__()
26
  self.mha = layers.MultiHeadAttention(num_heads=num_heads, key_dim=d_model)
27
- self.ffn = SwiGLU(d_model, dff)
28
  self.norm1 = layers.LayerNormalization(epsilon=1e-6)
29
  self.norm2 = layers.LayerNormalization(epsilon=1e-6)
30
  self.dropout1 = layers.Dropout(dropout)
 
24
  def __init__(self, d_model, num_heads, dff, dropout=0.1):
25
  super().__init__()
26
  self.mha = layers.MultiHeadAttention(num_heads=num_heads, key_dim=d_model)
27
+ self.ffn = SwiGLU(d_model, 512)
28
  self.norm1 = layers.LayerNormalization(epsilon=1e-6)
29
  self.norm2 = layers.LayerNormalization(epsilon=1e-6)
30
  self.dropout1 = layers.Dropout(dropout)