如何在 TensorFlow CNN 模型中初始化权重？

【问题标题】：How to initialize weights in tensorflow CNN model?如何在 TensorFlow CNN 模型中初始化权重？
【发布时间】：2018-11-09 22:15:03
【问题描述】：

我根据本教程构建了一个 CNN 模型： https://www.tensorflow.org/tutorials/layers

但是，我怀疑我的模型对权重的初始化很敏感，因为有时模型可以运行，但有时它会给出 NAN 错误。因此，我想初始化我选择的权重。这种情况下如何初始化权重？

【问题讨论】：

标签： python tensorflow machine-learning deep-learning convolutional-neural-network

【解决方案1】：

一般情况下，您需要在定义模型时定义初始化程序。对于罐装估算器，您无权访问它们的初始化程序。如果您定义自己的model_fn，则可以将kernel_initializer 和bias_initializer 用于tf.layers.conv2d 和tf.layers.dense。 Tensorflow 有一个tf.initializers module，它允许您定义自己的初始化程序。例如，如果您有特定的值，您可以使用tf.constant_initializer(init_value) 来初始化您选择的权重。

【讨论】：