gpt4 book ai didi

python - GRU加载模型错误,ValueError : GRU(reset_after=False) is not compatible with GRU(reset_after=True)

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

我是 ML 框架和 Python 的新手。我从 https://github.com/xiaochus/TrafficFlowPrediction 获得了 keras-tensorflow 项目的源代码而且我还安装了所有 CUDA 和 Cudnn 正确的版本。但在加载 gru 模型后会引发错误:

ValueError:GRU(reset_after=False) 与 GRU(reset_after=True) 不兼容。

谁能帮帮我?谢谢。似乎这个函数有一个带有选项的重载。我应该添加一些选项,如 reset_after 来启用/禁用它吗?我只是猜测。

lstm = load_model('TrafficFlowPrediction/model/lstm.h5')
#error in gru model load
gru = load_model('TrafficFlowPrediction/model/gru.h5')

最佳答案

我遇到了同样的问题,上面的解决方案对我不起作用。这就是我在这里添加它的原因。

我有一个包含 GRU 层的 TF1.x 保存模型。我想在 TF2.X 中转换我的代码,所以我遇到了上面同样的问题。

我用下面的代码解决了这个问题

tf.compat.v1.disable_v2_behavior() # model trained in tf1
model = tf.compat.v1.keras.models.load_model('models/model.h5')

关于python - GRU加载模型错误,ValueError : GRU(reset_after=False) is not compatible with GRU(reset_after=True),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/64324317/

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