From 520da2a1b93bdeba1d211c4b4a2d2dabc8de5648 Mon Sep 17 00:00:00 2001 From: CompatHelper Julia Date: Mon, 16 Dec 2024 01:45:28 +0000 Subject: [PATCH 1/3] CompatHelper: bump compat for Flux to 0.16 for package 2-deep-kernel-learning, (keep existing compat) --- examples/2-deep-kernel-learning/Project.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/2-deep-kernel-learning/Project.toml b/examples/2-deep-kernel-learning/Project.toml index 7ea6a335..cd91b34d 100644 --- a/examples/2-deep-kernel-learning/Project.toml +++ b/examples/2-deep-kernel-learning/Project.toml @@ -12,7 +12,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" [compat] AbstractGPs = "0.3,0.4,0.5" Distributions = "0.25" -Flux = "0.12, 0.13, 0.14" +Flux = "0.12, 0.13, 0.14, 0.16" KernelFunctions = "0.10" Literate = "2" MLDataUtils = "0.5" From 955b5fe057c86fb02b6b5e8a9c0b67c238eb4091 Mon Sep 17 00:00:00 2001 From: Hong Ge <3279477+yebai@users.noreply.github.com> Date: Wed, 28 May 2025 20:41:30 +0100 Subject: [PATCH 2/3] Update script.jl --- examples/2-deep-kernel-learning/script.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/2-deep-kernel-learning/script.jl b/examples/2-deep-kernel-learning/script.jl index 9ce11164..2ae3b3fe 100644 --- a/examples/2-deep-kernel-learning/script.jl +++ b/examples/2-deep-kernel-learning/script.jl @@ -71,13 +71,14 @@ plot!(vcat(x_test...), mean.(pred_init); ribbon=std.(pred_init), label="Predicti # ## Training nmax = 200 opt = Flux.Adam(0.1) +state = Flux.setup(opt, ps) anim = Animation() for i in 1:nmax grads = gradient(ps) do loss(y_train) end - Flux.Optimise.update!(opt, ps, grads) + Flux.Optimise.update!(state, ps, grads) if i % 10 == 0 L = loss(y_train) From 66f9e25a2d657ae4828ae85f151c4deb1707de00 Mon Sep 17 00:00:00 2001 From: Hong Ge <3279477+yebai@users.noreply.github.com> Date: Wed, 28 May 2025 20:41:57 +0100 Subject: [PATCH 3/3] Update Project.toml --- examples/2-deep-kernel-learning/Project.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/2-deep-kernel-learning/Project.toml b/examples/2-deep-kernel-learning/Project.toml index cd91b34d..50e0f948 100644 --- a/examples/2-deep-kernel-learning/Project.toml +++ b/examples/2-deep-kernel-learning/Project.toml @@ -12,7 +12,7 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" [compat] AbstractGPs = "0.3,0.4,0.5" Distributions = "0.25" -Flux = "0.12, 0.13, 0.14, 0.16" +Flux = "0.15, 0.16" KernelFunctions = "0.10" Literate = "2" MLDataUtils = "0.5"