├── README.md ├── __init__.py ├── energy_sampling ├── __init__.py ├── buffer.py ├── energies │ ├── __init__.py │ ├── base_set.py │ ├── easy_funnel.py │ ├── hard_funnel.py │ ├── many_well.py │ ├── nine_gmm.py │ └── twenty_five_gmm.py ├── evaluations.py ├── gflownet_losses.py ├── langevin.py ├── models │ ├── __init__.py │ ├── architectures.py │ └── gfn.py ├── plot_utils.py ├── sample_metrics.py ├── train.py └── utils.py ├── requirements.txt └── vae ├── __init__.py ├── buffer.py ├── energies ├── __init__.py ├── data │ ├── MNIST │ │ └── raw │ │ │ ├── t10k-images-idx3-ubyte │ │ │ ├── t10k-images-idx3-ubyte.gz │ │ │ ├── t10k-labels-idx1-ubyte │ │ │ ├── t10k-labels-idx1-ubyte.gz │ │ │ ├── train-images-idx3-ubyte │ │ │ ├── train-images-idx3-ubyte.gz │ │ │ ├── train-labels-idx1-ubyte │ │ │ └── train-labels-idx1-ubyte.gz │ ├── bce_mnist_vae_epochs_100_seed_1.pt │ ├── bernoulli_mnist_vae_epochs_100_seed_1.pt │ ├── mnist_evaluation_subset.npy │ └── mnist_vae_100_epochs_seed_1_mps.pt ├── vae.py ├── vae_energy.py └── vae_utils.py ├── evaluations.py ├── gflownet_losses.py ├── langevin.py ├── models ├── __init__.py ├── architectures.py └── gfn.py ├── plot_utils.py ├── sample_metrics.py ├── train.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /energy_sampling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /energy_sampling/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/buffer.py -------------------------------------------------------------------------------- /energy_sampling/energies/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/energies/__init__.py -------------------------------------------------------------------------------- /energy_sampling/energies/base_set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/energies/base_set.py -------------------------------------------------------------------------------- /energy_sampling/energies/easy_funnel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/energies/easy_funnel.py -------------------------------------------------------------------------------- /energy_sampling/energies/hard_funnel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/energies/hard_funnel.py -------------------------------------------------------------------------------- /energy_sampling/energies/many_well.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/energies/many_well.py -------------------------------------------------------------------------------- /energy_sampling/energies/nine_gmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/energies/nine_gmm.py -------------------------------------------------------------------------------- /energy_sampling/energies/twenty_five_gmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/energies/twenty_five_gmm.py -------------------------------------------------------------------------------- /energy_sampling/evaluations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/evaluations.py -------------------------------------------------------------------------------- /energy_sampling/gflownet_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/gflownet_losses.py -------------------------------------------------------------------------------- /energy_sampling/langevin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/langevin.py -------------------------------------------------------------------------------- /energy_sampling/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .gfn import * 2 | -------------------------------------------------------------------------------- /energy_sampling/models/architectures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/models/architectures.py -------------------------------------------------------------------------------- /energy_sampling/models/gfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/models/gfn.py -------------------------------------------------------------------------------- /energy_sampling/plot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/plot_utils.py -------------------------------------------------------------------------------- /energy_sampling/sample_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/sample_metrics.py -------------------------------------------------------------------------------- /energy_sampling/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/train.py -------------------------------------------------------------------------------- /energy_sampling/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/energy_sampling/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /vae/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/__init__.py -------------------------------------------------------------------------------- /vae/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/buffer.py -------------------------------------------------------------------------------- /vae/energies/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/__init__.py -------------------------------------------------------------------------------- /vae/energies/data/MNIST/raw/t10k-images-idx3-ubyte: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/data/MNIST/raw/t10k-images-idx3-ubyte -------------------------------------------------------------------------------- /vae/energies/data/MNIST/raw/t10k-images-idx3-ubyte.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/data/MNIST/raw/t10k-images-idx3-ubyte.gz -------------------------------------------------------------------------------- /vae/energies/data/MNIST/raw/t10k-labels-idx1-ubyte: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/data/MNIST/raw/t10k-labels-idx1-ubyte -------------------------------------------------------------------------------- /vae/energies/data/MNIST/raw/t10k-labels-idx1-ubyte.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/data/MNIST/raw/t10k-labels-idx1-ubyte.gz -------------------------------------------------------------------------------- /vae/energies/data/MNIST/raw/train-images-idx3-ubyte: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/data/MNIST/raw/train-images-idx3-ubyte -------------------------------------------------------------------------------- /vae/energies/data/MNIST/raw/train-images-idx3-ubyte.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/data/MNIST/raw/train-images-idx3-ubyte.gz -------------------------------------------------------------------------------- /vae/energies/data/MNIST/raw/train-labels-idx1-ubyte: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/data/MNIST/raw/train-labels-idx1-ubyte -------------------------------------------------------------------------------- /vae/energies/data/MNIST/raw/train-labels-idx1-ubyte.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/data/MNIST/raw/train-labels-idx1-ubyte.gz -------------------------------------------------------------------------------- /vae/energies/data/bce_mnist_vae_epochs_100_seed_1.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/data/bce_mnist_vae_epochs_100_seed_1.pt -------------------------------------------------------------------------------- /vae/energies/data/bernoulli_mnist_vae_epochs_100_seed_1.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/data/bernoulli_mnist_vae_epochs_100_seed_1.pt -------------------------------------------------------------------------------- /vae/energies/data/mnist_evaluation_subset.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/data/mnist_evaluation_subset.npy -------------------------------------------------------------------------------- /vae/energies/data/mnist_vae_100_epochs_seed_1_mps.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/data/mnist_vae_100_epochs_seed_1_mps.pt -------------------------------------------------------------------------------- /vae/energies/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/vae.py -------------------------------------------------------------------------------- /vae/energies/vae_energy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/vae_energy.py -------------------------------------------------------------------------------- /vae/energies/vae_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/energies/vae_utils.py -------------------------------------------------------------------------------- /vae/evaluations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/evaluations.py -------------------------------------------------------------------------------- /vae/gflownet_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/gflownet_losses.py -------------------------------------------------------------------------------- /vae/langevin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/langevin.py -------------------------------------------------------------------------------- /vae/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .gfn import * 2 | -------------------------------------------------------------------------------- /vae/models/architectures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/models/architectures.py -------------------------------------------------------------------------------- /vae/models/gfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/models/gfn.py -------------------------------------------------------------------------------- /vae/plot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/plot_utils.py -------------------------------------------------------------------------------- /vae/sample_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/sample_metrics.py -------------------------------------------------------------------------------- /vae/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/train.py -------------------------------------------------------------------------------- /vae/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GFNOrg/gfn-diffusion/HEAD/vae/utils.py --------------------------------------------------------------------------------