You could use forward hooks to get the activations without changing the model as described here.
1 Like
You could use forward hooks to get the activations without changing the model as described here.