C3M3 Assignment Exercise 1

I’m stuck with this error on Exercise 1. My output matches the expected output. Any ideas?

Failed test case: Failed forward pass test.
Expected: None
Got: Expected key_padded_mask.shape[0] to be 10, but got 2

Found the problem. That last part about batch_first is not meant to be taken literally.

# Define the encoder, by calling nn.TransformerEncoderLayer with the appropriate parameters. Do not forget th pass the dropout value! Include the parameter batch_first = None

2 Likes