繁体   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