Skip to content

Commit 3d40fbd

Browse files
another Lux change
1 parent e737613 commit 3d40fbd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/diffeqfluxtests.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ end
6969
prob_trueode = ODEProblem(trueODEfunc, u0, tspan)
7070
ode_data = Array(solve(prob_trueode, Tsit5(), saveat = tsteps))
7171

72-
dudt2 = Lux.Chain((x, p) -> x .^ 3,
72+
dudt2 = Lux.Chain(x -> x .^ 3,
7373
Lux.Dense(2, 50, tanh),
7474
Lux.Dense(50, 2))
7575
prob_neuralode = NeuralODE(dudt2, tspan, Tsit5(), saveat = tsteps)

0 commit comments

Comments
 (0)