Backward hook sometimes not called

Hi,

I would guess that this most likely happens because of interactions between views and inplace operations.
In general I would discourage using module backward hook. You can see in the doc a warning that their current behavior is not always as expected.

In your case, it would be more reliable to use t.register_hook() directly on the Tensors of interest. Note that you can do this inside your forward hook.