使用tf.contrib.layers.fully_connected在TensorFlow中正则化

Question

fully_connected有一个名为weights_regularizer的命名参数，它接受tf.contrib.layers提供的正则化参数，但是不清楚培训者能够找到这些正则化损失张量，或者如果我们必须手动将它们添加到总损失中。

有人可以提供一个使用fully_connected正则化的例子吗？

Answer 1

正则化器输出被添加到一个名为REGULARIZATION_LOSSES的集合中（层调用model_variable与正规化器）。

您必须自己添加REGULARIZATION_LOSSES的内容。要获得REGULARIZATION_LOSSES集合中的所有张量，请使用tf.get_collection(tf.GraphKeys.REGULARIZATION_LOSSES)。