Skip to content

Commit

Permalink
update ISTA step-size in example
Browse files Browse the repository at this point in the history
When using the preconditioner, the step size should be of order 1.
  • Loading branch information
KrisThielemans committed Jul 5, 2024
1 parent a95372f commit 2c66f91
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion main_ISTA.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down

0 comments on commit 2c66f91

Please sign in to comment.