Save state also for backward pass

Author: theHamsta

src/pyronn_torch/conebeam.py

@@ -85,15 +85,16 @@ def backward(self, projection_grad, state=None, *args):
             state.projection_multiplier, volume_grad, *state.volume_origin,
             *state.volume_spacing)
 
+        self.state = state
         if return_none:
             return volume_grad, None
         else:
             return volume_grad,
 
 
 class _BackwardProjection(torch.autograd.Function):
-    backward = _ForwardProjection.forward
-    forward = _ForwardProjection.backward
+    backward = staticmethod(_ForwardProjection.forward)
+    forward = staticmethod(_ForwardProjection.backward)
 
 
 class ConeBeamProjector: