Skip to content

Commit 5d458ad

Browse files
fix typo
1 parent e0cd6d0 commit 5d458ad

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

docs/src/examples/optimal_control/optimal_control.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,8 @@ of a local minimum. This looks like:
3737

3838
```@example neuraloptimalcontrol
3939
using Lux, ComponentArrays, OrdinaryDiffEq, Optimization, OptimizationOptimJL,
40-
OptimizationOptimisers, SciMLSensitivity, Zygote, Plots, Statistics, Random
40+
OptimizationOptimisers, SciMLSensitivity, Zygote, Plots, Statistics, Random,
41+
ForwardDiff
4142
4243
rng = Random.default_rng()
4344
tspan = (0.0f0, 8.0f0)
@@ -89,7 +90,7 @@ end
8990
# Setup and run the optimization
9091
9192
loss1 = loss_adjoint(θ)
92-
adtype = Optimization.AutoForward()
93+
adtype = Optimization.AutoForwardDiff()
9394
optf = Optimization.OptimizationFunction((x, p) -> loss_adjoint(x), adtype)
9495
9596
optprob = Optimization.OptimizationProblem(optf, θ)

0 commit comments

Comments
 (0)