尝试使用 keras 保存然后加载模型时出错：“ValueError：无法加载总共 1 个对象。”

Question

我正在尝试保存 DNN 模型，然后使用 keras 方法加载模型

load_model

。 以下是从Keras文档复制的简单示例。

import keras

model = keras.Sequential(
    [
        keras.layers.Dense(5, input_shape=(3,)),
        keras.layers.Softmax(),
    ],
)
model.save("model.keras")
loaded_model = keras.saving.load_model("model.keras")

不幸的是，在我的电脑上的 VS Code 中运行此代码会产生以下错误： ValueError: 无法加载总共 1 个对象。 还会显示随机选择的“示例错误消息”，但似乎并不特别有帮助。

如何在没有上述 ValueError 的情况下保存和加载模型，最好使用

.keras

扩展？

（还有其他几个松散相关的 SO 问题，例如 this 一个，但这些问题通常涉及遗留扩展

.h5

，而不是推荐的本机格式

.keras

。）

Answer 1

我在使用 keras 和 torch 后端时遇到了同样的问题。我对此很陌生，我不太清楚原因，但将后端更改为张量流为我解决了这个问题。 您可以在此处或此处获取有关如何更改 keras 后端的说明。 您还可以阅读文档这里。

希望有帮助。