GRU 加载模型错误，ValueError：GRU(reset_after=False)

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

Answer 2

我通过更改创建模型的这部分解决了这个问题model.add(GRU(单位[1], input_shape=(单位[0], 1), return_sequences=True))model.add（GRU（单位[1]，input_shape =（单位[0]，1），return_sequences = True，reset_after = True））