Solved by albanD in post #2 
              
              
                Hi, 


Hooks can allow you to get gradients wrt to intermediate results in the forward, not just the weights/biases. 


For backward hooks, you should only use the Tensor version right now (the nn.Module version is not working properly at the moment). 
To you a module hook for the output of cnn1 for…