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
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