Add initial step size choice (missed a line)

SyneRBI · Sep 13, 2024 · 0b442fb · 0b442fb
1 parent 6fdf0b2
commit 0b442fb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/main.py b/main.py
@@ -107,6 +107,7 @@ def __init__(self, data: Dataset):
         sampler = Sampler.random_without_replacement(len(obj_funs))
         f = -SVRGFunction(obj_funs, sampler=sampler, snapshot_update_interval=None, store_gradients=True)
         g = IndicatorBox(lower=0, accelerated=True) # non-negativity constraint
+        initial_step_size = initial_step_size_search_rule(data.OSEM_image, f, g, grad)
         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,