如何在TensorFlow中使用相同的权重进行初始化?
问题描述
我正在使用TensorFlow构建CNN模型,我想知道是否可以以某种方式设置权重初始化的种子,以便能够在任何地方都拥有相同的种子,以便可以在我的模型的不同体系结构之间进行比较。在堆栈溢出问题上找到了类似Tensorflow weight initialization的答案,但对我没有真正的帮助,因此我们将感谢您的帮助。提前感谢!
解决方案
您可以根据需要在所有层设置kernel_initializer
。术语kernel_initializer是一个奇特的术语,其统计分布或函数用于初始化权重。
如果您使用的是TensorFlow for Layer,则可以在此处找到不同初始化器的详细信息-Tensorflow Initialzer
如果您使用的是Kera for Layer,则可以在此处找到有关不同初始值设定项的详细信息-Keras Initializer
相关文章