change names

moskomule · Feb 4, 2023 · 6e9e8fd · 6e9e8fd
1 parent 791bd92
commit 6e9e8fd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/dataset_distillation.py b/examples/dataset_distillation.py
@@ -76,7 +76,7 @@ def outer_update(self) -> None:
  implicit_grads = self.approx_ihvp(lambda i, o: self.inner_obj(i, o, in_input, in_target)[0],
  lambda i, o: self.outer_obj(i, o, out_input, out_target)[0],
  in_params, out_params)
- self.outer_grad(implicit_grads)
+ self.outer_grad = implicit_grads
  self.outer_optimizer.step()
  self.outer.zero_grad(set_to_none=True)
 

diff --git a/examples/logistic_regression.py b/examples/logistic_regression.py
@@ -55,7 +55,7 @@ def outer_update(self) -> None:
  lambda i, o: self.outer_obj(i, o, out_input, out_target)[0],
  in_params, out_params)
 
- self.outer_grad(implicit_grads)
+ self.outer_grad = implicit_grads
  self.outer_optimizer.step()
  self.outer.zero_grad(set_to_none=True)