We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1f52339 commit 6d9ccb2Copy full SHA for 6d9ccb2
docs/src/examples/ode/second_order_neural.md
@@ -33,7 +33,7 @@ t = range(tspan[1], tspan[2], length = 20)
33
model = Chain(Dense(2, 50, tanh), Dense(50, 2))
34
ps, st = Lux.setup(Random.default_rng(), model)
35
ps = ComponentArray(ps)
36
-model = Lux.Experimental.StatefulLuxLayer(model, ps, st)
+model = StatefulLuxLayer{true}(model, ps, st)
37
38
ff(du, u, p, t) = model(u, p)
39
prob = SecondOrderODEProblem{false}(ff, du0, u0, tspan, ps)
0 commit comments