UserWarning: Using a non-full backward hook when the forward contains multiple autograd Nodes is deprecated and will be removed in future versions. This hook will be missing some grad_input. Please use register_full_backward_hook to get the documented behavior.

[slerman12]

Getting this warning:

UserWarning: Using a non-full backward hook when the forward contains multiple autograd Nodes is deprecated and will be removed in future versions. This hook will be missing some grad_input. Please use register_full_backward_hook to get the documented behavior.

[bhattg]

Hey can you describe more about this issue? I am also getting the same

[optimass]

same

[v-liuwei]

As said by the official documents,

This function(register_backward_hook()) is deprecated in favor of register_full_backward_hook() and the behavior of this function will change in future versions.

So just replace register_backward_hook() with register_full_backward_hook() in autograd_hacks.py at line 56.

autograd-hacks/autograd_hacks.py

Line 56 in 2c2e494

handles.append(layer.register_backward_hook(_capture_backprops))