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 loadgru = 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））