Open yumath opened 1 year ago
In PyTorch >= 1.8, this register_backward_hook can't be used directly, and register_full_backward_hook always return "using input grad before output grad bla...". Did you know how to replace register_backward_hook in higher PyTorch version?
register_backward_hook
register_full_backward_hook
In PyTorch >= 1.8, this
register_backward_hookcan't be used directly, andregister_full_backward_hookalways return "using input grad before output grad bla...". Did you know how to replaceregister_backward_hookin higher PyTorch version?