From 26db761406ed91226e830b0b3a8dd18e3bd7eaf1 Mon Sep 17 00:00:00 2001 From: "Sam.Porter" Date: Fri, 13 Sep 2024 16:55:18 +0100 Subject: [PATCH] typo from SVRG copy fixed --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index f341f77..55ac03c 100644 --- a/main.py +++ b/main.py @@ -88,11 +88,11 @@ def __init__(self, data: Dataset): sampler = Sampler.random_without_replacement(len(obj_funs)) preconditioner = MyPreconditioner(data.kappa) - f = -SAGAFunction(obj_funs, sampler=sampler, snapshot_update_interval=None, store_gradients=True) + f = -SAGAFunction(obj_funs, sampler=sampler) g = IndicatorBox(lower=0, accelerated=True) # non-negativity constraint step_size_rule = LinearDecayStepSizeRule(initial_step_size, decay=decay) super().__init__(initial=data.OSEM_image, f=f, g=g, step_size=step_size_rule, preconditioner=preconditioner, update_objective_interval=0) -submission_callbacks = [MaxIteration(np.inf)] \ No newline at end of file +submission_callbacks = [] \ No newline at end of file