I am not sure why this happens…
Maybe due to the partial that you use to give the name?
Also note that we currently discourage the use of the backward_hook() on the nn.Module (or at least be aware that they can return incorrect results) see the doc for more details.