I understand, but I want to avoid outputting a lot of intermediate activities from my forward function. Just like when we use the register_forward_hook on Modules, I wonder if there is an equivalent way to define resigter_hook on the parameters of a model.
Thanks,