From 6bb24de8d16ecf49ef19dbe3caa1665ce0ca484a Mon Sep 17 00:00:00 2001
From: Jens Lucht <jens.lucht@uni-goettingen.de>
Date: Wed, 8 Jan 2025 18:31:38 +0100
Subject: [PATCH] Add missing increase of niter for FISTA
---
src/hotopy/optimize.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/hotopy/optimize.py b/src/hotopy/optimize.py
index fe773ac..8bdb8e4 100644
--- a/src/hotopy/optimize.py
+++ b/src/hotopy/optimize.py
@@ -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
--
GitLab