Skip to content

Commit 36299df

Browse files
Fix regularization parameter initialization in LMSolver and R2NSolver
1 parent cc03f69 commit 36299df

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/LM_alg.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ function LMSolver(
6666
shifted(reg_nls.h, xk)
6767

6868
Jk = jac_op_residual(reg_nls.model, xk)
69-
sub_nlp = LMModel(Jk, Fk, T(1), xk)
69+
sub_nlp = LMModel(Jk, Fk, T(0), xk)
7070
subpb = RegularizedNLPModel(sub_nlp, ψ)
7171
substats = RegularizedExecutionStats(subpb)
7272
subsolver = subsolver(subpb)
@@ -300,7 +300,7 @@ function SolverCore.solve!(
300300
end
301301

302302
mk = let ψ = ψ, solver = solver
303-
d -> obj(solver.subpb.model, d) - 1/2 * solver.subpb.model.σ * dot(d, d) + ψ(d)
303+
d -> obj(solver.subpb.model, d) - solver.subpb.model.σ * dot(d, d) / 2 + ψ(d)
304304
end
305305

306306
prox!(s, ψ, mν∇fk, ν)

src/R2N.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ function R2NSolver(
6868
shifted(reg_nlp.h, xk)
6969

7070
Bk = hess_op(reg_nlp.model, x0)
71-
sub_nlp = R2NModel(Bk, ∇fk, T(1), x0)
71+
sub_nlp = R2NModel(Bk, ∇fk, T(0), x0)
7272
subpb = RegularizedNLPModel(sub_nlp, ψ)
7373
substats = RegularizedExecutionStats(subpb)
7474
subsolver = subsolver(subpb)
@@ -327,7 +327,7 @@ function SolverCore.solve!(
327327
end
328328

329329
mk = let ψ = ψ, solver = solver
330-
d -> obj(solver.subpb.model, d) - 1/2 * solver.subpb.model.σ * dot(d, d) + ψ(d)::T
330+
d -> obj(solver.subpb.model, d) - solver.subpb.model.σ * dot(d, d) / 2 + ψ(d)::T
331331
end
332332

333333
prox!(s1, ψ, mν∇fk, ν₁)

0 commit comments

Comments
 (0)