使用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)。