Add debug code

pomonam · Jul 8, 2024 · e0bc934 · e0bc934
1 parent 7c6e0a3
commit e0bc934
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/kronfluence/module/tracker/factor.py b/kronfluence/module/tracker/factor.py
@@ -273,6 +273,7 @@ def backward_hook(output_gradient: torch.Tensor) -> None:
 
         @torch.no_grad()
         def shared_backward_hook(output_gradient: torch.Tensor) -> None:
+            print("Shared processed")
             handle = self.cached_hooks.pop()
             handle.remove()
             output_gradient = self._preprocess_gradient(