Getting Intermediate Output of Self Created Sequential

vision
2

You could store the activations in a dict using forward hooks.
I’ve created a small example in this thread.

1 Like