Skip to content

Commit 0b2d218

Browse files
Get docs and tests building again
1 parent 24e9fde commit 0b2d218

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

docs/src/tutorials/chaotic_ode.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ u0 = [1.0, 0.0, 0.0]
3535
prob = ODEProblem(lorenz!, u0, tspan, p)
3636
sol = solve(prob, Vern9(), abstol = 1e-14, reltol = 1e-14)
3737
sol2 = solve(prob, Vern9(), abstol = 1e-14 + eps(Float64), reltol = 1e-14)
38-
pl1 = plot(sol, vars = (1, 2, 3), legend = true,
38+
pl1 = plot(sol, idxs = (1, 2, 3), legend = true,
3939
label = "sol",
4040
labelfontsize = 20,
4141
lw = 2,
@@ -122,7 +122,7 @@ g(u, p, t) = u[end]
122122
123123
function G(p)
124124
_prob = remake(prob_attractor, p = p)
125-
_sol = solve(_prob, Vern9(), abstol = 1e-14, reltol = 1e-14, saveat = 0.01,
125+
_sol = solve(_prob, Vern9(), abstol = 1e-8, reltol = 1e-8, saveat = 0.01,
126126
sensealg = ForwardLSS(g = g))
127127
sum(getindex.(_sol.u, 3))
128128
end
@@ -133,7 +133,7 @@ Alternatively, we can define the `ForwardLSSProblem` and solve it
133133
via `shadow_forward` as follows:
134134

135135
```@example chaosode
136-
sol_attractor = solve(prob_attractor, Vern9(), abstol = 1e-14, reltol = 1e-14)
136+
sol_attractor = solve(prob_attractor, Vern9(), abstol = 1e-8, reltol = 1e-8)
137137
lss_problem = ForwardLSSProblem(sol_attractor, ForwardLSS(g = g))
138138
resfw = shadow_forward(lss_problem)
139139
```

test/gpu/diffeqflux_standard_gpu.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ function predict_neuralode(p)
3030
end
3131
function loss_neuralode(p)
3232
pred = predict_neuralode(p)
33-
loss = sum(abs2, ode_data .- pred)
33+
loss = sum(abs2, ode_data .- gdev(pred))
3434
return loss
3535
end
3636

0 commit comments

Comments
 (0)