如何将 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 保存它。
有人知道如何将 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 保存它。