Laurent/Diffusers.jl
1
0
Fork 0
mirror of https://github.com/Laurent2916/Diffusers.jl.git synced 2024-10-17 23:26:19 +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.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