summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--RELEASES.md4
-rw-r--r--ot/backend.py2
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)