Remove self cache

pomonam · Jul 9, 2024 · 9744252 · 9744252
1 parent 6645ca9
commit 9744252
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kronfluence/module/tracker/pairwise_score.py b/kronfluence/module/tracker/pairwise_score.py
@@ -88,7 +88,7 @@ def backward_hook(output_gradient: torch.Tensor) -> None:
                     input_activation=cached_activation.to(device=output_gradient.device),
                     output_gradient=output_gradient,
                 )
-                del self.cached_activations, cached_activation, output_gradient
+                del cached_activation, output_gradient
                 self.clear_all_cache()
             else:
                 per_sample_gradient = self.module.compute_per_sample_gradient(