如何在张量流中用相同的权重初始化?

时间:2020-03-27 02:57:55

标签: python tensorflow neural-network conv-neural-network tensorflow2.0

我正在使用Tensorflow构建CNN模型,我想知道是否有可能以某种方式设置权重初始化的种子,以便能够在各处拥有相同的种子,以便可以在模型的不同体系结构之间进行比较。在Tensorflow weight initialization这样的堆栈溢出中找到了答案,但并没有真正帮助我,因此我们将不胜感激。预先感谢!

1 个答案:

答案 0 :(得分:1)

您可以根据需要在所有层中设置 kernel_initializer 。术语kernel_initializer是一个幻想术语,​​其统计分布或函数用于初始化权重。

初始化器可以采用不同的值。您可以将它们全部设置为零,一或恒定值。您还可以选择带有种子值的RandomUniform或RandomNormal。

如果您将tensorflow用于图层,则可以在此处找到有关不同初始化程序的详细信息-Tensorflow Initialzer

如果将Keras用于图层,则可以在此处找到有关不同初始化程序的详细信息-Keras Initializer