Skip to content

Commit

Permalink
Add the Manifest back
Browse files Browse the repository at this point in the history
  • Loading branch information
avik-pal committed Apr 27, 2024
1 parent c7161e4 commit e0c9464
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 1 deletion.
2 changes: 1 addition & 1 deletion Project.toml
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ LuxAMDGPU = "0.2.2"
LuxCUDA = "0.3.2"
LuxCore = "0.1.14"
LuxDeviceUtils = "0.1.19"
LuxLib = "0.3.15"
LuxLib = "0.3.18"
LuxTestUtils = "0.1.15"
MLUtils = "0.4.3"
MPI = "0.20.19"
Expand Down
4 changes: 4 additions & 0 deletions examples/NeuralODE/main.jl
Original file line number Diff line number Diff line change
Expand Up @@ -171,8 +171,12 @@ function train(model_function; cpu::Bool=false, kwargs...)
end

train(NeuralODECompact)
nothing #hide

#-

train(NeuralODE)
nothing #hide

# We can also change the sensealg and train the model! `GaussAdjoint` allows you to use
# any arbitrary parameter structure and not just a flat vector (`ComponentArray`).
Expand Down

0 comments on commit e0c9464

Please sign in to comment.