ptrblck:

you could also return the intermediate activations from some layers directly in your forward method (if you are using a custom model).


I think this suggestion will be my answer, but why didn’t I think of this?