Update AlphaS2S.py

AlphaS2S.py

@@ -201,10 +201,10 @@ class Transformer(tf.keras.Model):
         super().__init__()
         self.max_len = max_len
         self.d_model = d_model
-        self.enc_embedding = layers.Embedding(input_vocab_size, d_model)
-        self.enc_pos_embedding = layers.Embedding(max_len, d_model)
-        self.dec_embedding = layers.Embedding(target_vocab_size, d_model)
-        self.dec_pos_embedding = layers.Embedding(max_len, d_model)
+        self.enc_embedding = layers.Embedding(input_vocab_size, 256)
+        self.enc_pos_embedding = layers.Embedding(max_len, 256)
+        self.dec_embedding = layers.Embedding(target_vocab_size, 256)
+        self.dec_pos_embedding = layers.Embedding(max_len, 256)
         self.enc_layers = [EncoderBlock(d_model, num_heads, dff, dropout) for _ in range(num_layers)]
         self.dec_layers = [DecoderBlock(d_model, num_heads, dff, dropout) for _ in range(num_layers)]
         self.final_layer = layers.Dense(target_vocab_size, use_bias=False)