首页 > 解决方案 > 当我想在 PC 中加载该模型时,我可以在 colab bu 中加载我的深度模型,但我不能

问题描述

我通过 keras=2.3.1 和 tensorflow=2.1.0 在 colab 中训练了一个深度模型,我使用 JSON 和 Keras 保存了我的模型:

saveWeightPath = "/content/drive/My Drive/model_info/model_weights.h5"
saveModelPath = "/content/drive/My Drive/model_info"
model.save_weights(saveWeightPath)
tf.keras.models.save_model(model, saveModelPath)

和我保存 json 的成本

model_json = model.to_json()
with open("model.json", "w") as json_file:
json_file.write(model_json)
model.save_weights("model.h5")
print("Saved model to disk")

当我想在 colab 中加载模型时我没有问题,但是当我想在 pc local 中加载我的模型时我不能并且我得到这个错误:“

"

  1. 我的 Keras 和张量版本与我的 colab Keras 和张量版本相同

  2. 我尝试在 pc 中使用 JSON 和 Keras 加载我的模型

  3. 我收到一个回溯错误,我复制了错误消息的最后 5 行和此处的顶行 3

    最重要的错误行

标签: tensorflowkerasdeep-learning

解决方案


您好,首先您需要存储您的模型或模型重量吗?要知道这之间的区别,model.save()请保存权重和结构模型......但model.save_weights()只需保存您的权重模型,我建议您查看此链接以获取更多信息。

如果要保存模型,我建议使用model.save("test.hd5")ormodel.save(test.hdf5")和 usetensorflow.kears.models.load_model("test.hd5")来加载模型。


推荐阅读