Extract specific outputs through pre trained classifier

[ptrblck]

You could use forward hooks to get the intermediate activations from your model as described here. Let me know, if this would work for you or if I misunderstood the question.