Based on the error message the error is raised in self.fc1, which expects 16 * 224 * 224 = 802816 input features, while the incoming activation contains 150528 features.
You can fix it by setting in_features=150528 in self.fc1.