Skip to content

zmheiko/variational-perspective-on-gflownets

Repository files navigation

Code for the TMLR paper A Variational Perspective on Generative Flow Networks by Heiko Zimmermann, Fredrik Lindsten, Jan-Willem van de Meent, and Christian A Naesseth.

Requirements: All required packages are listed in req.pip and can be installed by running pip install -r req.pip.

Running experiments: Clone repository and run git submodule init followed by git submodule update. The submodule is needed for the synthetic density experiments and is a fork of the code of the ICML 2022 paper Generative Flow Networks for Discrete Probabilistic Modeling by Dinghuai Zhang, Nikolay Malkin, Zhen Liu, Alexandra Volokhova, Aaron Courville, and Yoshua Bengio.

  • Execute run_densities2d.sh to run the synthetic density experiments with the parameters specified in the script.
  • Execute run_ising.sh to run the Ising model experiments with the parameters specified in the script.

Inspecting results: Results (model parameters, evaluation metrics, and sample plots of final model) are saved in ./multiruns/<hostname>/<experiment_name>/<experiment_id_string>/ in ./params, ./eval, and ./plots respectively.

About

Code for the TMLR paper "A Variational Perspective on Generative Flow Networks"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published