Skip to content

Commit 13aaca7

Browse files
Update minibatch.md
1 parent 12f15fa commit 13aaca7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/src/examples/minibatch.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@ function dudt_(u,p,t)
2525
end
2626

2727
function predict_adjoint(time_batch)
28-
Array(concrete_solve(prob, Tsit5(),
29-
u0, θ, saveat = time_batch))
28+
_prob = remake(prob,u0=u0,p=θ)
29+
Array(solve(_prob, Tsit5(), saveat = time_batch))
3030
end
3131

3232
function loss_adjoint(batch, time_batch)

0 commit comments

Comments
 (0)