Solved by albanD in post #4 
              
              
                This is “expected” behaviour as stated in this comment in the code. 
For the record, 
In your case you trigger this because: 

The last Function on which you put the backward hook contains the backward hook.
This backward hook function because of the partial, references the parent nn.Module
This nn.…