DDPM PyTorch

Code Ref

• Deeplearning.ai(How Diffusion Model Work)
• ML simple works Blog

---

TODO

• sampling animation
• speed up sampling time
• conditional sampling

---

How To Use

# setup pytorch enviornment
# install library:
# pip install -r requirementx.txt

# train
python run.py --mode=trian

# sampling
python run.py --mode=sampling

# ddim sampling
python run.py --mode=ddim_sampling

# For more detail, see run.sh

---

Sampling Animation

Jupyter notebook code : here

DDPM

50 Epoch DDPM

30 Epoch DDPM

10 Epoch DDPM

DDPM with Context

Hero

Non-Hero

Food

spell&weapons

Side Facing

DDIM

DDPM : 2.939 sec (NVIDIA TITAN V)

DDIM : 0.981 sec (NVIDIA TITAN V)