backpropagator / Boundary-Preserving-Twin-EBMs

Repository for Boundary-Preserving Twin-EBMs

Boundary-Preserving-Twin-EBMs

Repository for Boundary-Preserving Twin-EBMs

Steps to run the code for BPT-EBM.

Train the VQ-VAE using run_ae.sh
Train the twin-EBMs using run_ebm_cyclic.sh

About

Repository for Boundary-Preserving Twin-EBMs

Languages

Language:Python 89.4%Language:Cuda 9.2%Language:C++ 1.1%Language:Shell 0.4%