wenxinxu/resnet-in-tensorflow

function create_variables in resnet.py

Nina-9519 opened this issue · 1 comments


def create_variables(name, shape, initializer=tf.contrib.layers.xavier_initializer(), is_fc_layer=False):
if is_fc_layer is True:
regularizer = tf.contrib.layers.l2_regularizer(scale=FLAGS.weight_decay)
else:
regularizer = tf.contrib.layers.l2_regularizer(scale=FLAGS.weight_decay)

new_variables = tf.get_variable(name, shape=shape, initializer=initializer,
                                regularizer=regularizer)
return new_variables

These two lines, the same?

Ah thanks for pointing out! I intended to use different weight--decay for fc layers.

Changed now.