Hi torch experts,
I am trying to add a forward hook to my model. But I got the error message indicating the hook won’t work with jit module.
Is there a particular reason why hook won’t work with jit modules? How can I work around this while still keep my module as a script module?
Thanks in advance!