From 7fa22f8c665c6b66b8b72e1eee382cb13665a3d7 Mon Sep 17 00:00:00 2001 From: Kris Thielemans Date: Fri, 5 Jul 2024 09:56:25 +0100 Subject: [PATCH] update ISTA step_size Now that we're using the preconditioner, the step-size should be of order 1. --- main_ISTA.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main_ISTA.py b/main_ISTA.py index e24138f..3afe121 100644 --- a/main_ISTA.py +++ b/main_ISTA.py @@ -48,7 +48,7 @@ class Submission(ISTA): """Stochastic subset version of preconditioned ISTA""" # note that `issubclass(ISTA, Algorithm) == True` - def __init__(self, data: Dataset, num_subsets: int = 7, step_size: float = 1e-6, + def __init__(self, data: Dataset, num_subsets: int = 7, step_size: float = 0.1, update_objective_interval: int = 10): """ Initialisation function, setting up data & (hyper)parameters.