reset initial step size after armijo step

SyneRBI · Sep 30, 2024 · 9f8dbee · 9f8dbee
1 parent f893385
commit 9f8dbee
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -147,6 +147,8 @@ def get_step_size(self, algorithm):
             # Update the internal state with the new step size as the minimum of the current and previous step sizes
             self.step_size = min(step_size, self.step_size)
 
+            self.initial_step_size = self.step_size
+
             if self.counter < self.steps:
                 self.counter += 1