Add Tutorial Oriented Repo MNISTDiffusion: A Lightweight DDPM With Minimal Code and Dependencies.

diff-usion / Awesome-Diffusion-Models

A collection of resources and papers on Diffusion Models

https://diff-usion.github.io/Awesome-Diffusion-Models/

MIT License

10.84k stars 934 forks source link

Add Tutorial Oriented Repo MNISTDiffusion: A Lightweight DDPM With Minimal Code and Dependencies. #117

Open bot66 opened 9 months ago

bot66 commented 9 months ago

Hi!

I would like to contribute my repo to this awesome curated list. In this repo I implemented a denoising diffusion model from scratch with merely PyTorch.The model is surprisingly small and works perfectly fine on the MNIST dataset. You can train it on a CPU machine at a reasonable speed.

I also include the good resources I have checked when building the model in the README. This is a very good repo for beginners who are looking for a simple codebase of DDPM.