It is a similar issue to before; the input shape to the first linear layer must be [?, 32] according to the current code. You can compute (or simply print the shape before the linear layer) to see what the shape of the linear layer should be changed to. However, I don’t see a call to the linear layer in your forward function so I’m not sure if the code posted corresponds to the error.
The docs for linear layer 4 show a good example of how the input shape should be matched to the linear layer.