Awesome-Diffusion-Models icon indicating copy to clipboard operation
Awesome-Diffusion-Models copied to clipboard

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

Open bot66 opened this issue 1 year ago • 0 comments

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.

bot66 avatar Dec 14 '23 02:12 bot66