This may be the simplest implement of DDPM. I trained with CIFAR-10 dataset. The links of pretrain weight, which trained on CIFAR-10 are in the Issue 2.
HOW TO RUN
-
- You can run Main.py to train the UNet on CIFAR-10 dataset. After training, you can set the parameters in the model config to see the amazing process of DDPM.
-
- You can run MainCondition.py to train UNet on CIFAR-10. This is for DDPM + Classifier free guidence.
Some generated images are showed below:
-
- DDPM without guidence:
-
- DDPM + Classifier free guidence: