Hi
I used that function on ReLUs, and faced this error.
ModuleAttributeError: ‘LeakyReLU’ object has no attribute ‘register_full_backward_hook’
I’m using dgl with PyTorch Backend, so could it be that it doesn’t recognize this function?