Skip to content
forked from cvignac/DiGress

code for the paper "DiGress: Discrete Denoising diffusion for graph generation"

License

Notifications You must be signed in to change notification settings

sriramdvt/DiGress

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Denoising diffusion models for graph generation

Warning 1: when running the code, you might encounter an AttributeError cls._old_init. This is a non deterministic error due to a bad interaction between pytorch_lightning and torch_geometric. Just run the code again until it works (it might happen up to 5 times in a row)

Warning 2: The code has been updated since experiments were run for the paper. If you don't manage to reproduce the paper results, please write to us so that we can investigate the issue.

Warning 3: the conditional generation experiments were implemented with an legacy version of the code. They are not yet available in the public version.

Environment installation

Download the data

Run the code

  • All code is currently launched through python3 main.py. Check hydra documentation (https://hydra.cc/) for overriding default parameters.
  • To run the debugging code: python3 main.py +experiment=debug.yaml. We advise to try to run the debug mode first before launching full experiments.
  • To run a code on only a few batches: python3 main.py general.name=test.
  • To run the continuous model: python3 main.py model=continuous
  • To run the discrete model: python3 main.py
  • You can specify the dataset with python3 main.py dataset=guacamol. Look at configs/dataset for the list of datasets that are currently available

Cite the paper

@article{vignac2022digress,
  title={DiGress: Discrete Denoising diffusion for graph generation},
  author={Vignac, Clement and Krawczuk, Igor and Siraudin, Antoine and Wang, Bohan and Cevher, Volkan and Frossard, Pascal},
  journal={arXiv preprint arXiv:2209.14734},
  year={2022}
}

About

code for the paper "DiGress: Discrete Denoising diffusion for graph generation"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 70.2%
  • C++ 29.8%