Skip to content

enijkamp/triangle

Repository files navigation

Divergence Triangle for Joint Training of Generator Model, Energy-based Model, and Inference Model.

Publication.

https://arxiv.org/abs/1812.10907.

@article{han2018divergence,
  title={Divergence Triangle for Joint Training of Generator Model, Energy-based Model, and Inference Model},
  author={Han, Tian and Nijkamp, Erik and Fang, Xiaolin and Hill, Mitch and Zhu, Song-Chun and Wu, Ying Nian},
  journal={arXiv preprint arXiv:1812.10907},
  year={2018}
}

Training on CIFAR-10.

pip3 install -r requirements.txt
python3 train_cifar10.py