From 6dfaa8b0d632d1141a981cac608b8b6c6489dafd Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Thu, 25 Apr 2024 23:17:54 -0400 Subject: [PATCH] Add the Manifest back --- Project.toml | 4 ++-- examples/NeuralODE/main.jl | 4 ++++ 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/Project.toml b/Project.toml index 86f6d428d..de2c5e6b1 100644 --- a/Project.toml +++ b/Project.toml @@ -65,7 +65,7 @@ ADTypes = "0.2, 1" Adapt = "4" Aqua = "0.8.4" ArrayInterface = "7.9" -CUDA = "5.2" +CUDA = "5.3.2" ChainRulesCore = "1.23" ComponentArrays = "0.15.11" ConcreteStructs = "0.2.3" @@ -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" diff --git a/examples/NeuralODE/main.jl b/examples/NeuralODE/main.jl index 2901534a9..5dc0acb36 100644 --- a/examples/NeuralODE/main.jl +++ b/examples/NeuralODE/main.jl @@ -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`).