Replace register_backward_hook with register_full_backward_hook

Summary: register_backward_hook is deprecated. Differential Revision: D68562558
pytorch · Jan 23, 2025 · a9c8e84 · a9c8e84
1 parent ea4cb95
commit a9c8e84
Showing 1 changed file with 1 addition and 1 deletion.
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,