首页 > 解决方案 > Keras 符号输入/输出不实现 `op`

问题描述

我正在关注本教程,试图将 .h5 模型转换为 tensorRT 模型 链接到教程

我正在使用 tensorflow 2.6.0,所以我更改了一些行,但我被困在模型冻结块中。我在这些行中有这个错误:

input_names = [t.op.name for t in model.inputs] output_names = [t.op.name for t in model.outputs]

两行的问题都是一样的:

TypeError:Keras 符号输入/输出未实现op。您可能正在尝试将 Keras 符号输入/输出传递给未注册调度的 TF API,从而阻止 Keras 自动将 API 调用转换为功能模型中的 lambda 层。

正如我所说,我正在关注本教程,但我有一个昨天训练的高效NetB5模型,我认为如果我将它加载到笔记本的这一部分并开始冻结它就足够了,但现在我可以不要继续。

知道这里发生了什么吗?

标签: pythontensorflowkeras

解决方案


推荐阅读