This error is raised in the output_layer, as the incoming number of features is not matching the defined in_features in this layer.
How did you define the in_features?
If the model was working before, did you change the spatial size of your input?
Also, is your batch size 16? If not, a view or reshape operation might have been called in a wrong way inside your model.