diff --git a/kronfluence/module/tracker/pairwise_score.py b/kronfluence/module/tracker/pairwise_score.py index 2bde117..a04239f 100644 --- 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(