【发布时间】:2018-09-26 03:53:24
【问题描述】:
有人知道如何将 tensorflow keras 检查点文件保存到 .pb 二进制文件中以供服务吗?
我不是 100% 确定如何实现这一点。
是否需要声明tf.keras.backend.get_session() 才能保存图形和pb 文件?
总之,
1.通过tf.keras保存pb文件的方法
或者
2. 任何使用tf.keras.estimator.model_to_estimator(keras_model=model)的方法
谢谢
【问题讨论】:
标签: tensorflow keras tensorflow-datasets tensorflow-estimator