gpt4 book ai didi

tensorflow - TF2.0创建单个pb文件没有额外的文件夹或文件?

转载 作者:行者123 更新时间:2023-12-04 04:21:24 24 4
gpt4 key购买 nike

我搜索并阅读了几篇文章,但找不到任何提示来创建没有变量、 Assets 文件夹的单个 pb 文件。

我的 tensorflow 版本是 2.0.0

https://www.tensorflow.org/guide/saved_model

从这个文档中,我可以从检查点创建一个 pb 文件。
我在我的文件夹中找到了一个具有这种结构的 pb 文件。

  • pb_file_parent_dir/ Assets
  • pb_file_parent_dir/saved_model.pb
  • pb_file_parent_dir/变量/

  • 我想创建一个 pb 文件(saved_model.pb),但我有 2 个额外的文件夹( Assets 和变量)。

    如果有办法创建单个 pb 文件,请帮帮我。我想为部署目的创建一个 pb 文件。

    已编辑

    我尝试使用 'tf.keras.models.save_model' 和 https://www.tensorflow.org/api_docs/python/tf/keras/Model#save .我得到了与上面相同的结果,而不是单个 pb 文件。

    感谢您有一个愉快的一天。

    最佳答案

    TensorFlow 2.0 不再使用这种“卡住”的单个文件 .pb 来存储模型,而是使用 SavedModel 格式 https://www.tensorflow.org/guide/saved_model (这是您已经拥有的 .pb 和 variables/assets 文件夹)或者您可以使用 Keras .h5 格式 https://www.tensorflow.org/tutorials/keras/save_and_load#hdf5_format .

    您仍然可以使用 compat 模块 https://www.tensorflow.org/api_docs/python/tf/compat/v1/graph_util/convert_variables_to_constants 在 tf2.0 中创建卡住的 .pb 文件。但是你必须使用所有兼容的东西,比如 session (我相信这不是 TensorFlow 2.0 的做事方式,并且与 keras api 不兼容)。

    关于tensorflow - TF2.0创建单个pb文件没有额外的文件夹或文件?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59206573/

    24 4 0
    Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
    广告合作:1813099741@qq.com 6ren.com