This may be the simplest implement of DDPM. You can directly run Main.py to train the UNet on CIFAR-10 dataset and see the amazing process of denoising.
ddpm
Kennzahlen
Sterne-Wachstum
Sterne
2.2k
Forks
220
Wochenwachstum
—
Issues
35
1k2k
Juni 2022Feb. 2023Okt. 2023Juni 2024März 2025Nov. 2025Juli 2026
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.
If you really want to know more about the framwork of DDPM, I have listed some papers for reading by order in the closed Issue 1.
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.