Update AlphaS2S.py

AlphaS2S.py

@@ -203,7 +203,7 @@ class Transformer(tf.keras.Model):
         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, 256)
+        self.dec_embedding = layers.Embedding(target_vocab_size, d_model)
         self.dec_pos_embedding = layers.Embedding(max_len, d_model)
         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)]