Как можно использовать регуляризацию в TensorFlow-Slim?

Question

Я хочу использовать регуляризацию в моем коде. Я использовал slim для создания conv2d следующим образом:

slim.conv2d(input, 256, [1, 1], stride=1,  padding='SAME', scope='conv1')

Как я могу добавить регуляризацию к этому? и как я могу использовать его для упорядочения моей потери?

Answer 1

Да просто можно просто добавить параметр

weights_regularizer=slim.l2_regularizer(0.001)

или вы можете использовать slim.arg_scope, чтобы установить регуляризацию для нескольких слоев:

with slim.arg_scope([slim.conv2d],
                    padding='SAME',
                    weights_regularizer=slim.l2_regularizer(0.001)):
      net = slim.conv2d(input, 256, [1, 1], scope='conv1')
      net = slim.conv2d(net, 256, [1, 1], scope='conv2')