我是 ML 框架和 python 的新手。我從https://github.com/xiaochus/TrafficFlowPrediction獲取了 keras-tensorflow 項(xiàng)目的源代碼，并且還安裝了所有 CUDA 和 Cudnn 正確版本。但加載 gru 模型后會(huì)引發(fā)錯(cuò)誤：ValueError：GRU（reset_after = False）與GRU（reset_after = True）不兼容。有人可以幫我嗎？謝謝?？磥?lái)這個(gè)函數(shù)的選項(xiàng)有重載。我應(yīng)該添加一些選項(xiàng)（例如reset_after）來(lái)啟用/禁用它嗎？我只是猜測(cè)。lstm = load_model('TrafficFlowPrediction/model/lstm.h5')#error in gru model loadgru = load_model('TrafficFlowPrediction/model/gru.h5')
2 回答
慕田峪9158850
TA貢獻(xiàn)1794條經(jīng)驗(yàn) 獲得超8個(gè)贊
我遇到了同樣的問(wèn)題，上面的解決方案對(duì)我不起作用。這就是我在這里添加這個(gè)的原因。
我有一個(gè)包含 GRU 層的 TF1.x 保存模型。我想將我的代碼轉(zhuǎn)換為 TF2.X，所以我遇到了上面同樣的問(wèn)題。
我通過(guò)以下代碼解決了這個(gè)問(wèn)題
tf.compat.v1.disable_v2_behavior() # model trained in tf1
model = tf.compat.v1.keras.models.load_model('models/model.h5')
慕勒3428872
TA貢獻(xiàn)1848條經(jīng)驗(yàn) 獲得超6個(gè)贊
我通過(guò)更改創(chuàng)建模型的這部分解決了這個(gè)問(wèn)題
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））
添加回答
舉報(bào)
0/150
提交
取消