Replace register_backward_hook with register_full_backward_hook (#720)

iden-kalemaj · facebook-github-bot · commit 47421993495f · 2025-01-27T07:37:15.000-08:00
Summary:

register_backward_hook is deprecated.

Differential Revision: D68562558
diff --git a/opacus/grad_sample/grad_sample_module.py b/opacus/grad_sample/grad_sample_module.py
@@ -207,7 +207,7 @@ def add_hooks(
             )
 
             self.autograd_grad_sample_hooks.append(
-                module.register_backward_hook(
+                module.register_full_backward_hook(
                     partial(
                         self.capture_backprops_hook,
                         loss_reduction=loss_reduction,

Original file line number	Diff line number	Diff line change
`@@ -207,7 +207,7 @@ def add_hooks(`
`207`	`207`	`)`
`208`	`208`
`209`	`209`	`self.autograd_grad_sample_hooks.append(`
`210`		`- module.register_backward_hook(`
	`210`	`+ module.register_full_backward_hook(`
`211`	`211`	`partial(`
`212`	`212`	`self.capture_backprops_hook,`
`213`	`213`	`loss_reduction=loss_reduction,`