It seems the number of features in the incoming activation tensor doesn’t match the expected in_features in the linear layer.
Could you post the code for the model definition as well as the input shapes?