The Identity assignment would be a hack in case you don’t want to manipulate the forward.
Since you are changing it, just comment out the calls to self.avgpool and self.fc1 and check the shape of the output after passing some random input to the model.