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