Skip to content

Commit e33c9e4

Browse files
Fix minibatch example
1 parent e1c5b04 commit e33c9e4

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docs/src/tutorials/minibatch.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,11 @@
22

33
!!! note
44

5-
This example uses the OptimizationOptimJL.jl package. See the [Optim.jl page](@ref optim)
5+
This example uses the OptimizationOptimisers.jl package. See the [Optimisers.jl page](@ref optimisers)
66
for details on the installation and usage.
77

88
```julia
9-
using DiffEqFlux, Optimization, OptimizationOptimJL, OrdinaryDiffEq
9+
using DiffEqFlux, Optimization, OptimizationOptimisers, OrdinaryDiffEq
1010

1111
function newtons_cooling(du, u, p, t)
1212
temp = u[1]
@@ -65,6 +65,6 @@ l1 = loss_adjoint(pp, train_loader.data[1], train_loader.data[2])[1]
6565
optfun = OptimizationFunction((θ, p, batch, time_batch) -> loss_adjoint(θ, batch, time_batch), Optimization.AutoZygote())
6666
optprob = OptimizationProblem(optfun, pp)
6767
using IterTools: ncycle
68-
res1 = Optimization.solve(optprob, ADAM(0.05), ncycle(train_loader, numEpochs), callback = callback)
68+
res1 = Optimization.solve(optprob, Optimisers.ADAM(0.05), ncycle(train_loader, numEpochs), callback = callback)
6969
@test 10res1.minimum < l1
7070
```

0 commit comments

Comments
 (0)