GRU加载模型错误，ValueError:GRU（reset\u after=False）与GRU（reset\u after=True）不兼容

Question

我是ML框架和python的新手。我从https://github.com/xiaochus/TrafficFlowPrediction获得了keras tensorflow项目的源代码，并且安装了所有CUDA和Cudnn正确的版本。但在加载gru模型后，会产生一个错误：

值错误：GRU（reset\u after=False）与GRU（reset\u 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

我通过改变创建模型的这一部分解决了这个问题

add（GRU（单位[1]，输入_形状=（单位[0]，1]，返回_序列=真））

添加（GRU（单位[1]，输入单位形状=（单位[0]，1），返回单位序列=真，重置单位后=真））