From 6e9e8fd41ae3a246e199b556e03dfcbe0a5f05f9 Mon Sep 17 00:00:00 2001 From: mosko Date: Sat, 4 Feb 2023 13:30:07 +0900 Subject: [PATCH] change names --- examples/dataset_distillation.py | 2 +- examples/logistic_regression.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/dataset_distillation.py b/examples/dataset_distillation.py index be6f72e..911d877 100644 --- 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 index 54015b6..a81e785 100644 --- 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)