Skip to content

Commit

Permalink
changeD SVRG to SAGA
Browse files Browse the repository at this point in the history
  • Loading branch information
samdporter committed Sep 13, 2024
1 parent 92f66c3 commit a42e8a5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
"""
#%%
from cil.optimisation.algorithms import ISTA, Algorithm
from cil.optimisation.functions import IndicatorBox, SVRGFunction
from cil.optimisation.functions import IndicatorBox, SAGAFunction
from cil.optimisation.utilities import (Preconditioner, Sampler,
callbacks, StepSizeRule)
from petric import Dataset
Expand Down Expand Up @@ -88,7 +88,7 @@ def __init__(self, data: Dataset):

sampler = Sampler.random_without_replacement(len(obj_funs))
preconditioner = MyPreconditioner(data.kappa)
f = -SVRGFunction(obj_funs, sampler=sampler, snapshot_update_interval=None, store_gradients=True)
f = -SAGAFunction(obj_funs, sampler=sampler, snapshot_update_interval=None, store_gradients=True)
g = IndicatorBox(lower=0, accelerated=True) # non-negativity constraint
step_size_rule = LinearDecayStepSizeRule(initial_step_size, decay=decay)

Expand Down

0 comments on commit a42e8a5

Please sign in to comment.