Fix bug in solve_batch_cg

aai-institute · Oct 23, 2023 · 354774f · 354774f
1 parent 536c841
commit 354774f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pydvl/influence/torch/torch_differentiable.py b/src/pydvl/influence/torch/torch_differentiable.py
@@ -568,7 +568,7 @@ def solve_batch_cg(
     if len(training_data) == 0:
         raise ValueError("Training dataloader must not be empty.")
 
-    total_grad_xy = torch.empty()
+    total_grad_xy = torch.empty(0)
     total_points = 0
 
     for x, y in maybe_progress(training_data, progress, desc="Batch Train Gradients"):