Hi,
Do you have the python stack trace? This happens because you give matrices of the wrong size to a matrix multiply most likely.
You can add some prints during your forward pass to check if the Tensors are of the size you expect.