From 160b6df10228454bccd59b3b78547030c59bb312 Mon Sep 17 00:00:00 2001 From: Christopher Rackauckas Date: Wed, 7 Jul 2021 07:54:59 -0400 Subject: [PATCH] Update Kolmogorov equations --- tutorials/advanced/03-kolmogorov_equations.jmd | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tutorials/advanced/03-kolmogorov_equations.jmd b/tutorials/advanced/03-kolmogorov_equations.jmd index 2cf8ec6e..ab3cde11 100644 --- a/tutorials/advanced/03-kolmogorov_equations.jmd +++ b/tutorials/advanced/03-kolmogorov_equations.jmd @@ -82,15 +82,15 @@ Now lets define our model m and the optimiser ```julia m = Chain(Dense(d, 64, elu),Dense(64, 128, elu),Dense(128 , 16 , elu) , Dense(16 , 1)) use_gpu = false -if CUDAnative.functional() == true - m = fmap(CuArrays.cu , m) +if CUDA.functional() == true + m = fmap(CUDA.cu , m) use_gpu = true end opt = Flux.ADAM(0.0005) ``` And then finally call the solver ```julia -@time sol = solve(prob, NeuralNetDiffEq.NNKolmogorov(m, opt, sdealg, ensemblealg), verbose = true, dt = dt, +@time sol = solve(prob, NeuralPDE.NNKolmogorov(m, opt, sdealg, ensemblealg), verbose = true, dt = dt, dx = dx , trajectories = trajectories , abstol=1e-6, maxiters = 1000 , use_gpu = use_gpu) ``` ## Analyzing the solution