Skip to content

Commit

Permalink
Update SciMLSensitivity compats
Browse files Browse the repository at this point in the history
  • Loading branch information
avik-pal committed May 14, 2024
1 parent f0d2020 commit 8e2723c
Show file tree
Hide file tree
Showing 5 changed files with 8 additions and 6 deletions.
2 changes: 2 additions & 0 deletions docs/run_single_tutorial.jl
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,8 @@ output_directory = ARGS[2]
path = ARGS[3]

io = open(pkg_log_path, "w")
Pkg.Registry.update()
Pkg.update()
Pkg.develop(; path=joinpath(@__DIR__, ".."), io)
Pkg.instantiate(; io)
close(io)
Expand Down
2 changes: 1 addition & 1 deletion examples/GravitationalWaveForm/Project.toml
Original file line number Diff line number Diff line change
Expand Up @@ -25,4 +25,4 @@ LuxCUDA = "0.2, 0.3"
Optimization = "3"
OptimizationOptimJL = "0.1, 0.2"
OrdinaryDiffEq = "6"
SciMLSensitivity = "7"
SciMLSensitivity = "7.57"
2 changes: 1 addition & 1 deletion examples/NeuralODE/Project.toml
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,6 @@ MLUtils = "0.2, 0.3, 0.4"
OneHotArrays = "0.1, 0.2"
Optimisers = "0.2, 0.3"
OrdinaryDiffEq = "6"
SciMLSensitivity = "7.45"
SciMLSensitivity = "7.57"
Statistics = "1"
Zygote = "0.6"
4 changes: 2 additions & 2 deletions examples/PolynomialFitting/main.jl
Original file line number Diff line number Diff line change
Expand Up @@ -74,12 +74,12 @@ vjp_rule = AutoZygote()
function main(tstate::Lux.Experimental.TrainState, vjp, data, epochs)
data = data .|> gpu_device()
for epoch in 1:epochs
grads, loss, stats, tstate = Lux.Training.compute_gradients(
grads, loss, stats, tstate = Lux.Experimental.compute_gradients(
vjp, loss_function, data, tstate)
if epoch % 50 == 1 || epoch == epochs
@printf "Epoch: %3d \t Loss: %.5g\n" epoch loss
end
tstate = Lux.Training.apply_gradients!(tstate, grads)
tstate = Lux.Experimental.apply_gradients!(tstate, grads)
end
return tstate
end
Expand Down
4 changes: 2 additions & 2 deletions examples/SymbolicOptimalControl/Project.toml
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ Optimization = "3.24.3"
OptimizationOptimJL = "0.2.3"
OptimizationOptimisers = "0.2.1"
OrdinaryDiffEq = "6.74.1"
SciMLSensitivity = "7.56.2"
Statistics = "1.11.1"
SciMLSensitivity = "7.57"
Statistics = "1.11"
SymbolicRegression = "0.24.1"
SymbolicUtils = "1.5.1"

0 comments on commit 8e2723c

Please sign in to comment.