From 062134d39494a664668eb00d10f8364dff3c642e Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 10 Apr 2020 22:05:33 +0200
Subject: [PATCH] Save state also for backward pass
---
src/pyronn_torch/conebeam.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/pyronn_torch/conebeam.py b/src/pyronn_torch/conebeam.py
index c9409fb..846ef63 100644
--- a/src/pyronn_torch/conebeam.py
+++ b/src/pyronn_torch/conebeam.py
@@ -85,6 +85,7 @@ class _ForwardProjection(torch.autograd.Function):
state.projection_multiplier, volume_grad, *state.volume_origin,
*state.volume_spacing)
+ self.state = state
if return_none:
return volume_grad, None
else:
@@ -92,8 +93,8 @@ class _ForwardProjection(torch.autograd.Function):
class _BackwardProjection(torch.autograd.Function):
- backward = _ForwardProjection.forward
- forward = _ForwardProjection.backward
+ backward = staticmethod(_ForwardProjection.forward)
+ forward = staticmethod(_ForwardProjection.backward)
class ConeBeamProjector:
--
GitLab