如何在张量流CNN模型中初始化权重？

Question

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

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

Answer 1

通常，在定义模型时需要定义初始化程序。对于预设估算器，您无法访问其初始化程序。如果您定义自己的model_fn，则可以使用kernel_initializer bias_initializer tf.layers.conv2d和tf.layers.dense。 Tensorflow有tf.initializers module，允许您定义自己的初始化程序。例如，如果您有特定的值，则可以使用tf.constant_initializer(init_value)来初始化您选择的权重。