GRU 加载模型错误，ValueError：GRU(reset_after=False) 与 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')