mirror of
https://github.com/wassname/denoising-diffusion-pytorch.git
synced 2026-09-10 12:01:08 +08:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
edc6d5136a | ||
|
|
207d23af86 | ||
|
|
cf4f44909f | ||
|
|
a7f2d670bb |
@@ -431,7 +431,7 @@ class GaussianDiffusion(nn.Module):
|
|||||||
beta_schedule = 'cosine',
|
beta_schedule = 'cosine',
|
||||||
p2_loss_weight_gamma = 0., # p2 loss weight, from https://arxiv.org/abs/2204.00227 - 0 is equivalent to weight of 1 across time - 1. is recommended
|
p2_loss_weight_gamma = 0., # p2 loss weight, from https://arxiv.org/abs/2204.00227 - 0 is equivalent to weight of 1 across time - 1. is recommended
|
||||||
p2_loss_weight_k = 1,
|
p2_loss_weight_k = 1,
|
||||||
ddim_sampling_eta = 1.
|
ddim_sampling_eta = 0.
|
||||||
):
|
):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
assert not (type(self) == GaussianDiffusion and model.channels != model.out_dim)
|
assert not (type(self) == GaussianDiffusion and model.channels != model.out_dim)
|
||||||
|
|||||||
@@ -414,7 +414,7 @@ class GaussianDiffusion1D(nn.Module):
|
|||||||
beta_schedule = 'cosine',
|
beta_schedule = 'cosine',
|
||||||
p2_loss_weight_gamma = 0.,
|
p2_loss_weight_gamma = 0.,
|
||||||
p2_loss_weight_k = 1,
|
p2_loss_weight_k = 1,
|
||||||
ddim_sampling_eta = 1.
|
ddim_sampling_eta = 0.
|
||||||
):
|
):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
self.model = model
|
self.model = model
|
||||||
|
|||||||
@@ -1 +1 @@
|
|||||||
__version__ = '0.1.0'
|
__version__ = '0.1.5'
|
||||||
|
|||||||
Reference in New Issue
Block a user