PaddleEBM

This repository is a PaddlePaddle Library of Energy-based Models (EBMs).

Prerequisites

• Linux or macOS
• Python 3.6+
• PaddlePaddle 2.0+

Getting Started

• Clone this repo:

git clone https://github.com/zilongzheng/PaddleEBM.git
cd PaddleEBM

• Install Python dependencies by

pip install -r requirements.txt

Train

• Train a model (e.g. CoopNets on MNIST dataset):

python train.py --config-file configs/coopnets_mnist.yaml

Publication

• Patchwise Generative ConvNet: Training Energy-Based Models from a Single Natural Image for Internal Learning. CVPR 2021 [Pdf][Model][Config]
• Learning Cycle-Consistent Cooperative Networks via Alternating MCMC Teaching for Unsupervised Cross-Domain Translation. AAAI 2021
• Generative VoxelNet: Learning Energy-Based Models for 3D Shape Synthesis and Analysis. TPAMI 2020 [Pdf]
• Cooperative Training of Descriptor and Generator Networks. TPAMI 2018 [Pdf][Model]

Acknowledgements

This repository is based on PaddleGAN and PaddlePaddle official implementation.