Laurent/Diffusers.jl
1
0
Fork 0
mirror of https://github.com/Laurent2916/Diffusers.jl.git synced 2024-10-18 07:36:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

⚡️ (examples/swissroll) switch beta schedule to cosine

Browse source
This commit is contained in:
Laureηt 2023-08-15 15:48:15 +02:00
parent c665be0dd5
commit 2a0b019deb
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
examples/swissroll.jl
View file

@ -1,7 +1,7 @@
import Diffusers import Diffusers
import Diffusers.Schedulers import Diffusers.Schedulers
import Diffusers.Schedulers: DDPM import Diffusers.Schedulers: DDPM
import Diffusers.BetaSchedules: linear_beta_schedule import Diffusers.BetaSchedules: cosine_beta_schedule
using Flux using Flux
using Random using Random
using Plots using Plots
@ -38,7 +38,7 @@ scatter(dataset[1, :], dataset[2, :],
num_timesteps = 100 num_timesteps = 100
scheduler = DDPM( scheduler = DDPM(
linear_beta_schedule(num_timesteps) cosine_beta_schedule(num_timesteps)
); );
data = dataset data = dataset