in class DecoderLayer(tf.keras.layers.Layer): CLass
BLOCK 2
# calculate self-attention using the Q from the first block
#and K and V from the encoder output.
# Return attention scores as attn_weights_block2 (~1 line)
attn2, attn_weights_block2 = self.mha2(enc_output,enc_output, out1, padding_mask, return_attention_scores=True)
I get this error when I run the code any ideas? TIA
AssertionError: Wrong values in attn_w_b2. Check the call to self.mha2