Skip to content

Commit

Permalink
test: adjust learning rate
Browse files Browse the repository at this point in the history
  • Loading branch information
avik-pal committed Oct 4, 2024
1 parent b48eb92 commit 0323a3f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/callback_reversediff.jl
Original file line number Diff line number Diff line change
Expand Up @@ -61,6 +61,6 @@ cb1 = function (p, l)
end

res = solve(OptimizationProblem(OptimizationFunction(loss_n_ode, AutoZygote()), ps),
Adam(0.05); callback = cb1, maxiters = 100)
Adam(0.005); callback = cb1, maxiters = 250)

@test loss_n_ode(res.u, nothing) < 0.4

0 comments on commit 0323a3f

Please sign in to comment.