Skip to content

Commit 053bbe4

Browse files
committed
fix to method in ekfac
1 parent 4bfdcb6 commit 053bbe4

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/pydvl/influence/torch/influence_function_model.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1197,7 +1197,6 @@ def _solve_hvp(self, rhs: torch.Tensor) -> torch.Tensor:
11971197
return x
11981198

11991199
def to(self, device: torch.device):
1200-
self.model.to(device)
12011200
if self.is_fitted:
12021201
self.ekfac_representation.to(device)
1203-
return self
1202+
return super().to(device)

0 commit comments

Comments
 (0)