Solved! Check this: C4W2 Assignment DecoderLayer - #4 by Lidia_Opiola. If the testing function in the notebook works, please check if you use the right variable especially within next_word function.
1 Like