gpt4 book ai didi

python - 如何加载保存为 .pb 的 keras 模型

转载 作者:行者123 更新时间:2023-12-04 00:16:48 31 4
gpt4 key购买 nike

我想加载一个我训练过的 keras 模型并将其保存为 .pb .这是代码,
enter image description here
我正在使用 jupyter 笔记本。
模型成功保存为 save_model.pb 同目录下。但是代码无法访问它。
任何人都可以看到它,我如何访问保存在 中的这个 keras 模型.pb 延期。我在其他几个地方检查了解决方案,但没有运气。

模型保存在 模型/saved_model.pb .
我已经取出了 .pb 文件并将其放在我的代码文件所在的同一目录中。

最佳答案

函数tf.keras.models.load_model加载 SavedModeltf.keras -模型。该函数的参数是保存模型的路径。
所以试试 model = tf.keras.models.load_model('model')

关于python - 如何加载保存为 .pb 的 keras 模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63146892/

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