Skip to content

Commit ed62022

Browse files
committed
default to sigmoid noise schedule, seems to be a lot better
1 parent 7897ddb commit ed62022

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

denoising_diffusion_pytorch/denoising_diffusion_pytorch_1d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -411,7 +411,7 @@ def __init__(
411411
sampling_timesteps = None,
412412
loss_type = 'l1',
413413
objective = 'pred_noise',
414-
beta_schedule = 'cosine',
414+
beta_schedule = 'sigmoid',
415415
p2_loss_weight_gamma = 0.,
416416
p2_loss_weight_k = 1,
417417
ddim_sampling_eta = 0.
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = '0.1.7'
1+
__version__ = '0.2.2'

0 commit comments

Comments
 (0)