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

Question

我是 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')

Answer 1

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

我有一个包含 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')