tensorflowkerastensorflow-datasetstensorflow-estimator

how to save Tensorflow Keras checkpoint to pb file

Anyone knows how to save tensorflow keras checkpoint file into .pb binary file for serving?

I am not 100% sure how to implement this.

Do I need to declare tf.keras.backend.get_session() to save graph and pb file?

In sum, 1. Method to save pb file via tf.keras or 2. Any approaches using tf.keras.estimator.model_to_estimator(keras_model=model)

Thanks

Solution

  • Here is a link on how to use keras with tensorflow. Basically you should get the tensorflow model from keras (K.get_session() and K.get_graph().graph) and use the tensorflow methotd to save it.