Phil Wang e0f26677d6 make sure predicted mean is actually detached for all of the kl loss calculations

2022-05-12 11:12:34 -07:00

..

__init__.py

first pass at ddpm with learned variance

2022-05-11 17:38:29 -07:00

denoising_diffusion_pytorch.py

complete the gaussian diffusion with hybrid loss (learned variance) as in the improved ddpm paper

2022-05-12 08:54:47 -07:00

learned_gaussian_diffusion.py

make sure predicted mean is actually detached for all of the kl loss calculations

2022-05-12 11:12:34 -07:00