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

Question

6

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

有人能提供使用fully_connected进行正则化的示例吗？

- Cristian Garcia

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- wicke · Accepted Answer

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

您需要自己将REGULARIZATION_LOSSES的内容添加到您的损失中。要获取REGULARIZATION_LOSSES集合中的所有张量，请使用tf.get_collection(tf.GraphKeys.REGULARIZATION_LOSSES)。