簡體   English   中英

設置神經網絡的權重?

[英]Set the weights of a neural networks?

我現在正在編寫一個 DRL 算法,該算法使用雙 DQN 原理和一個 model 和一個隨時間更新的目標。

所以在我的代碼中我 go 這一行:

self.q_next.set_weights(self.q_eval.get_weights())

這會導致以下錯誤消息:

ValueError:您在“d3qn_1”層上調用set_weights(weights) ,權重列表長度為 10,但該層期望權重為 0。 提供的權重:[array([[[[ 0.04574016, 0.03492326, -0.04824715, ...

為什么我不能將一個網絡的權重設置為另一個?

我通過首先構建 q_next model 解決了這個問題,但是您需要提前知道輸入的維度。

例如:

self.target_model.build((None, 80, 80, 1))
self.target_model.set_weights(self.model.get_weights())

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM