How to get intermediate features from pre-trained model

vision
2

If you need conv2 output, yes it would work.

Another way is to override the forward function of the model class in a subclass and run the input through the layers (conv1, conv2) only.