Save TensorFlow subclass model


I am working on an image caption model but I faced a problem with saving the model as tflite or HDF5. I tried a lot of things but does not work.

My model is a subclass model and similar to this work :

Can anyone help me?

submitted by /u/Rama_AI
[visit reddit] [comments]

Leave a Reply

Your email address will not be published. Required fields are marked *