Skip to content

Latest commit

 

History

History
19 lines (14 loc) · 586 Bytes

README.md

File metadata and controls

19 lines (14 loc) · 586 Bytes

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