function create_variables in resnet.py
Nina-9519 opened this issue · 1 comments
Nina-9519 commented
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?
wenxinxu commented
Ah thanks for pointing out! I intended to use different weight--decay for fc layers.
Changed now.