Reproducing 3-digit MNIST BO task with decoder uncertainty

[PraljakReps]

Hello @pascalnotin,

Excellent paper and repository; however, I noticed that this repository only contains code on the junction tree VAE (JTVAE).

Is there reproducible code for the more straightforward task (MNIST 3-digit generation) which was presented in the paper: Improving black-box optimization in VAE latent space using decoder uncertainty (Pascal Notin, José Miguel Hernández-Lobato, Yarin Gal)?

[pascalnotin]

Hi @PraljakReps,
Thanks for the kind words! Will try and upload another task to the repo when I get the chance (post NeurIPS).
Best,
Pascal