Thanks for sharing the code.
However, it seems the Tensorflow VAE doesn't have implementation for color images, eg. cifar10 data, only MNIST and fashion MNIST is available.Also, the Pytorch version doesn't have VAE implementation at all.
Is there anyway to use VAE for color mages. Thanks.