From 2a0b019debc8c13185b4b9d651ecbe32212777e3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Laure=CE=B7t?= Date: Tue, 15 Aug 2023 15:48:15 +0200 Subject: [PATCH] =?UTF-8?q?=E2=9A=A1=EF=B8=8F=20(examples/swissroll)=20swi?= =?UTF-8?q?tch=20beta=20schedule=20to=20cosine?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- examples/swissroll.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/swissroll.jl b/examples/swissroll.jl index 4fe775e..1fdfd43 100644 --- a/examples/swissroll.jl +++ b/examples/swissroll.jl @@ -1,7 +1,7 @@ import Diffusers import Diffusers.Schedulers import Diffusers.Schedulers: DDPM -import Diffusers.BetaSchedules: linear_beta_schedule +import Diffusers.BetaSchedules: cosine_beta_schedule using Flux using Random using Plots @@ -38,7 +38,7 @@ scatter(dataset[1, :], dataset[2, :], num_timesteps = 100 scheduler = DDPM( - linear_beta_schedule(num_timesteps) + cosine_beta_schedule(num_timesteps) ); data = dataset