【问题标题】:how to save Tensorflow Keras checkpoint to pb file如何将 Tensorflow Keras 检查点保存到 pb 文件
【发布时间】: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


    【解决方案1】:

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

    【讨论】:

    • 我已经看过这篇文章,但是当我保存 keras 模型时,它需要通过 keras 模型重新加载,但我不知道该怎么做。以下说明对 pb 保存文件没有帮助。这里有没有简单的例子 tf.keras save pb 例子?
    猜你喜欢
    • 1970-01-01
    • 2018-01-09
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2020-04-10
    • 2019-09-29
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多