Skip to content
Snippets Groups Projects

Add missing increase of niter for FISTA

Merged Jens Lucht requested to merge fix-fista into master
1 file
+ 2
0
Compare changes
  • Side-by-side
  • Inline
+ 2
0
@@ -315,6 +315,8 @@ class FISTA:
@torch.no_grad()
def step(self):
self.niter += 1
# forward step: gradient decent step (note: gradients in y!)
self.y_hat = self.y - self.tau * self.y.grad
Loading