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 .
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}
}