Welcome to Diffusion World !
This repository is a collection of diffusion models,
implemented in PyTorch
and PyTorch Lightning
.
-
Models You Can Explore:
-
Datasets Integrated:
- To train
DDPM
withMNIST
dataset
python run.py --model=ddpm --dataset_name=mnist
- To train
D3PM
withCIFAR-10
dataset
python run.py --model=d3pm --discrete=True --dataset_name=cifar10
After training, retrieve your models and results from the ./results directory.
- For additional command line arguments, use:
python run.py --help
- DDPM with cifar10 dataset
python run.py --model_name=ddpm --dataset_name=cifar10
- D3PM with mnist dataset
python run.py --model_name=d3pm --discrete=True --dataset_name=mnist --scheduler_mode=jsd
Click to expand
-
Add a diffusion model of
Noise Conditional Score Network
(https://arxiv.org/abs/1907.05600) -
Add DDIM sampling method
Thanks to the following repositories and papers for the inspiration and guidance: