Hi,
The hook are only called for the part of the model that you backpropagate through.
And if you run the model in no grad mode, it will not compute gradients for that part and break the link between the input and output. So the gradient won’t flow backward past that part.