Skip to content

backpropagator/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