Skip to content

Commit 51ec2c2

Browse files
Make DiffEqFlux test more robust
1 parent bf83617 commit 51ec2c2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/diffeqfluxtests.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -99,13 +99,13 @@ prob = Optimization.OptimizationProblem(optprob, pp)
9999

100100
result_neuralode = Optimization.solve(prob,
101101
OptimizationOptimisers.ADAM(), callback = callback,
102-
maxiters = 300)
102+
maxiters = 500)
103103
@test result_neuralode.objectiveloss_neuralode(result_neuralode.u)[1] rtol=1e-2
104104

105105
prob2 = remake(prob, u0 = result_neuralode.u)
106106
result_neuralode2 = Optimization.solve(prob2,
107107
BFGS(initial_stepnorm = 0.0001),
108108
callback = callback,
109-
maxiters = 100)
109+
maxiters = 300)
110110
@test result_neuralode2.objectiveloss_neuralode(result_neuralode2.u)[1] rtol=1e-2
111111
@test result_neuralode2.objective < 10

0 commit comments

Comments
 (0)