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

时间:2018-09-26 03:53:25

标签: tensorflow keras tensorflow-datasets tensorflow-estimator

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

我不确定100%如何实现这一目标。

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

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

的方法

谢谢

1 个答案:

答案 0 :(得分:0)

这里是有关如何使用keras with tensorflow的链接,基本上,您应该从keras获取tensorflow模型(K.get_session()和K.get_graph()。graph),然后使用tensorflow methotd保存它。