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.