diff options
-rw-r--r-- | RELEASES.md | 4 | ||||
-rw-r--r-- | ot/backend.py | 2 |
2 files changed, 5 insertions, 1 deletions
diff --git a/RELEASES.md b/RELEASES.md index 9b92d97..c6ab9c3 100644 --- a/RELEASES.md +++ b/RELEASES.md @@ -6,6 +6,10 @@ - Better list of related examples in quick start guide with `minigallery` (PR #334) +#### Closed issues + +- Bug in instantiating an `autograd` function (`ValFunction`, Issue #337, PR #338) + ## 0.8.1.0 *December 2021* diff --git a/ot/backend.py b/ot/backend.py index 58b652b..6e0bc3d 100644 --- a/ot/backend.py +++ b/ot/backend.py @@ -1397,7 +1397,7 @@ class TorchBackend(Backend): def set_gradients(self, val, inputs, grads): - Func = self.ValFunction() + Func = self.ValFunction res = Func.apply(val, grads, *inputs) |