Skip to content

bpucla/latent-space-EBM-prior

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Learning Latent Space Energy-Based Prior Model

Code for reproducing reported results in Learning Latent Space Energy-Based Prior Model (https://arxiv.org/pdf/2006.08205.pdf)

Requirements:

tensorflow-gpu=1.14.0
torchvision=0.5.0
pytorch=1.4.0
scipy=1.1.0
scikit-learn=0.21.2
Pillow=6.2.0
matplotlib=3.1.1

Training:

python train_svhn.py

Citation

If you found this code useful please cite our paper

@article{pang2020ebmprior,
  title={Learning Latent Space Energy-Based Prior Model},
  author={Pang, Bo and Han, Tian and Nijkamp, Erik and Zhu, Song-Chun and Wu, Ying Nian},
  journal={NeurIPS 2020},
  year={2020}
}

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages