Tensorflow probability module in production

13 views
Skip to first unread message

Tridib dutta

unread,
May 10, 2022, 6:37:14 PMMay 10
to TensorFlow Probability
Hello all,

I have a model based on TFP (a variational autoencoder). I want to use a JAVA program to read in the trained model saved as .h5 file. However, while loading the saved model, I am having issues with some of the layers (KLDivergenceAddLoss) to be precise. I am wondering if there is a work around or should we avoid TFP to build the model in the first place? 
Any suggestion would be greatly appreciated. 

Thank you.
Reply all
Reply to author
Forward
0 new messages