Skip to content
master
Switch branches/tags
Go to file
Code

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
src
 
 
 
 
 
 
 
 

README.md

All in the Exponential Family: Bregman Duality in Thermodynamic Variational Inference

Code for All in the Exponential Family: Bregman Duality in Thermodynamic Variational Inference

@inproceedings{brekelmans2020allin,
  title     = {All in the Exponential Family: Bregman Duality in Thermodynamic Variational Inference},
  author    = {Rob Brekelmans and Vaden Masrani and Frank Wood and Greg Ver Steeg and Aram Galstyan},
  booktitle = {International Conference on Machine Learning},
  year      = {2020},
}

To download datasets, run scripts in data/ folder

Example run script:

python3 main.py with loss='tvo' schedule='moments' dataset='mnist' S=50 K=10 batch_size=1000 epochs=1000 seed=2 record=True verbose=True
loss in ['tvo', 'tvo_reparam', 'iwae', 'iwae_dreg', 'elbo']
schedule in ['moments', 'coarse_grain', 'log', 'linear']
dataset in ['mnist', 'omniglot']

About

No description, website, or topics provided.

Resources

Releases

No releases published

Packages

No packages published

Languages