complete training, sans sampling

This commit is contained in:
Phil Wang
2020-09-05 23:41:02 -07:00
parent e496bba350
commit ee76497b83
2 changed files with 80 additions and 4 deletions
+1 -1
View File
@@ -1 +1 @@
from denoising_diffusion_pytorch.denoising_diffusion_pytorch import DenoisingDiffusion, Unet
from denoising_diffusion_pytorch.denoising_diffusion_pytorch import GaussianDiffusion, Unet