mirror of
https://github.com/Laurent2916/Diffusers.jl.git
synced 2024-10-17 23:26:19 +00:00
⚡️ (examples/swissroll) switch beta schedule to cosine
This commit is contained in:
parent
c665be0dd5
commit
2a0b019deb
|
@ -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
|
||||||
|
|
Loading…
Reference in a new issue