wassname/denoising-diffusion-pytorch
1
0
Fork 0
mirror of https://github.com/wassname/denoising-diffusion-pytorch.git synced 2026-09-10 12:01:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

get working version of gaussian diffusion with continuous time (only beta linear schedule for now, but will eventually contain alpha cosine schedule as well as parameterized, learned monotonic MLP)

Browse Source
This commit is contained in:
Phil Wang
2022-06-07 15:59:29 -07:00
parent 6b93fa48f6
commit 3bbb6ebf16
5 changed files with 203 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
denoising_diffusion_pytorch/__init__.py
+1
View File
@@ -1,4 +1,5 @@
from denoising_diffusion_pytorch.denoising_diffusion_pytorch import GaussianDiffusion, Unet, Trainer
from denoising_diffusion_pytorch.learned_gaussian_diffusion import LearnedGaussianDiffusion
from denoising_diffusion_pytorch.continuous_time_gaussian_diffusion import ContinuousTimeGaussianDiffusion
from denoising_diffusion_pytorch.weighted_objective_gaussian_diffusion import WeightedObjectiveGaussianDiffusion