Skip to content

Commit

Permalink
Workaround for IndicatorBox leading to crash with accelerated=True
Browse files Browse the repository at this point in the history
  • Loading branch information
paskino committed Oct 17, 2024
1 parent 4e8dd56 commit a131cd9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion main.py
Original file line number Diff line number Diff line change
Expand Up @@ -246,7 +246,7 @@ def __init__(self, data: Dataset, update_objective_interval=10):
f = -FullGradientInitialiserFunction(obj_funs, sampler=sampler, init_steps=5)

preconditioner = BSREMPreconditioner(obj_funs, epsilon=data.OSEM_image.max()/1e6, freeze_iter=10*update_interval+5)
g = IndicatorBox(lower=0, accelerated=True) # non-negativity constraint
g = IndicatorBox(lower=0, accelerated=False) # non-negativity constraint

step_size_rule = ArmijoStepSearchRule(0.08, beta, decay, max_iter=100, tol=0.2, init_steps=5, update_interval=10*update_interval+5)

Expand Down

0 comments on commit a131cd9

Please sign in to comment.