The unit test for next_symbol was updated on 01/19, using the Lab Help button, I got the latest version, but it still seems to be incorrect. I don’t know how many others are doing this assignment after the update. Here is my question - w1_unittest.test_next_symbol(next_symbol, NMTAttn) is passing the function NMTAttn, not the model. Where am I supposed to point it to the trained model’s weights? I am also stuck on this same problem.
To answer my own question, there is no need to initialize weights. When I initialized weights inside next_symbol method, my output didn’t match the unit test’s expectation. I was calling NMTAttn like this - NMTAttn(params). Changing it to NMTAttn((params)) fixed this issue for me.