如何将 Tensorflow Keras 检查点保存到 pb 文件

how to save Tensorflow Keras checkpoint to pb file

有人知道如何将 tensorflow keras 检查点文件保存为 .pb 二进制文件以供服务吗?

我不是 100% 确定如何实现它。

我需要声明 tf.keras.backend.get_session() 来保存图形和 pb 文件吗?

总而言之, 1.通过tf.keras保存pb文件的方法 要么 2. 任何使用 tf.keras.estimator.model_to_estimator(keras_model=model)

的方法

谢谢

这里是 link 如何使用 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 保存它。