I’m not sure, why you don’t get the right activations.
The model definition looks a bit strange. If the model is designed to run as a sequential model, the shapes of the linear layers don’t match together.
In my code example the hooks run fine. Could you check, if you used the same code logic?