Solved by albanD in post #2 
              
              
                Hi, 
Are you using Module.register_backward_hook()? 
If so, this is a known issue and if you check the latest doc, they have been deprecated in favor of Module.register_full_backward_hook() that has the exact same API and doc (but actually do what the doc says they’re supposed to do  ).