From ffd1b9d7c3e75c1d90f0d1edc74268372a11abf3 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 28 Feb 2020 17:56:17 +0100
Subject: [PATCH] fix requires_grad
---
src/pystencils_autodiff/backends/_torch_native.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/src/pystencils_autodiff/backends/_torch_native.py b/src/pystencils_autodiff/backends/_torch_native.py
index 848da23..fa5c299 100644
--- a/src/pystencils_autodiff/backends/_torch_native.py
+++ b/src/pystencils_autodiff/backends/_torch_native.py
@@ -73,6 +73,9 @@ def create_autograd_function(autodiff_obj, use_cuda):
output_tensors = OrderedDict({f.name:
field_to_tensor_dict.get(f, kwargs[f.name])
for f in autodiff_obj.forward_output_fields})
+ for o in output_tensors.values():
+ if isinstance(o, torch.Tensor):
+ o.requires_grad = True
field_to_tensor_dict.update(kwargs)
self.saved_for_backward = kwargs
--
GitLab