Yuchan
commited on
Update AlphaS2S.py
Browse files- 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,
|
| 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)
|