Problem with NMT with Attention

I am having problem passing the unit tests for next_symbol. I have spent many hours staring at this problem and trying all sorts of workarounds. Could one of the moderators or students offer suggestions. Everything up to this point has been ok except for having to WeightedCategoryCrossEntropy to CrossEntropyLoss. Please advise. Thanks.

Please message me your Assignment notebook, I will take a look at it.

I’ve had many many issues with the next_symbol function. Everything went away by restarting the notebook.