Diffusion Normalizing Flow (DiffFlow)

Reproduce

setup environment

The repo heavily depends on jam, a personal toolbox developed by Qsh.zh. The API may change and check the jammy version for running the repo.

pip

pip install .

poetry

curl -fsS -o /tmp/get-poetry.py https://raw.githubusercontent.com/sdispater/poetry/master/get-poetry.py
python3 /tmp/get-poetry.py -y --no-modify-path
export PATH=$HOME/.poetry/bin:$PATH
poetry shell
poetry install

Run

python main.py trainer.epochs=100 data.dataset=tree

The repo supports viz results on wandb

python main.py trainer.epochs=100 data.dataset=tree log=true wandb.project=pub_diff wandb.name=tree

There are some results reproduced by the repo.

Reference

@inproceedings{zhang2021diffusion,
  author    = {Qinsheng Zhang and Yongxin Chen},
  title     = {Diffusion Normalizing Flow},
  booktitle = {Advances in Neural Information Processing Systems},
  year      = {2021}
}