我是 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')
2 回答
慕田峪9158850
TA貢獻1794條經驗 獲得超7個贊
我遇到了同樣的問題，上面的解決方案對我不起作用。這就是我在這里添加這個的原因。
我有一個包含 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')
慕勒3428872
TA貢獻1848條經驗 獲得超6個贊
我通過更改創建模型的這部分解決了這個問題
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））
添加回答
舉報
0/150