From 2c66f91bc51411bf5bb673c4c74bfdc6de40c4f0 Mon Sep 17 00:00:00 2001 From: Kris Thielemans Date: Fri, 5 Jul 2024 09:50:48 +0100 Subject: [PATCH] update ISTA step-size in example When 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 7c3dec2..881cd0f 100644 --- a/main_ISTA.py +++ b/main_ISTA.py @@ -47,7 +47,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.