OK, great! That’s how the error should have been reported in the first place.
As the error claims, you are running into a shape mismatch in a linear layer, so print the shape of the intermediate activations in the forward method and make sure the in_feautres of the crashing linear layer match the activation features.