Solved by albanD in post #2 
              
              
                Hi, 
First of all you do not need to explicitly register the parameters. This is done for you automatically when you save an nn.Parameter as self.foo. This means that you can remove your register_hook function completely and it won’t change anything. 
The problem here is that the operations that you…