Update AlphaS2S.py

AlphaS2S.py

@@ -208,7 +208,7 @@ class Transformer(tf.keras.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, d_model)
+        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)