Skip to content

kb107/Boundary-Preserving-Twin-EBMs

 
 

Repository files navigation

DOI

Boundary-Preserving-Twin-EBMs

Repository for Boundary-Preserving Twin-EBMs

Steps to run the code for BPT-EBM.

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

About

Repository for Boundary-Preserving Twin-EBMs

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 89.3%
  • Cuda 9.2%
  • C++ 1.1%
  • Shell 0.4%