HI,

Thank you for sharing your code.

However, when I run the plain VAE code using the default parameters, I cannot reproduce the result you post on the REAMED page.

The reconstruction image and random image I generated using your code is:

Can you provide some suggestions on how to reproduce your result?