Skip to content

Commit 12f15fa

Browse files
Update stiff_ode_fit.md
1 parent ee4937f commit 12f15fa

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

docs/src/examples/stiff_ode_fit.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,8 @@ Js = map(u->I + 0.1*ForwardDiff.jacobian(UJacobianWrapper(rober, 0.0, p), u), so
3030

3131
function predict_adjoint(p)
3232
p = exp.(p)
33-
Array(concrete_solve(prob,Rosenbrock23(autodiff=false),u0,p,saveat=ts,sensealg=QuadratureAdjoint(autojacvec=ReverseDiffVJP(true))))
33+
_prob = remake(prob,p=p)
34+
Array(solve(_prob,Rosenbrock23(autodiff=false),saveat=ts,sensealg=QuadratureAdjoint(autojacvec=ReverseDiffVJP(true))))
3435
end
3536

3637
function loss_adjoint(p)

0 commit comments

Comments
 (0)