Based on the error message and that you were using flatten = 1800 before, I guess the shape mismatch is created in the self.linear module of the Private module.
Assuming the [8 x 11552] shape represents the activations using the 224x224 inputs, set flatten to 11552 and rerun your script.