How to Save Model that has a TensorFlow Probability Regularizer?
One solution I found was saving the model in HD5 format. That seems to bypass the serialization of the regularizer:
vae.save(&quot;./VAE.h5&quot;)
You can also try saving the weights only: vae.save_weights(&quot;./VAE-weights&quot;)