Skip to content

Commit

Permalink
update ISTA step_size
Browse files Browse the repository at this point in the history
Now that we're using the preconditioner, the step-size should be
of order 1.
  • Loading branch information
KrisThielemans committed Jul 5, 2024
1 parent c45b39b commit 7fa22f8
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 @@ -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.
Expand Down

0 comments on commit 7fa22f8

Please sign in to comment.