in utils.py [0:0]
def init_conv_weight(
weights,
scope,
k,
c_in,
c_out,
spec_norm=True,
zero=False,
scale=1.0,
classes=1):
if spec_norm:
spec_norm = FLAGS.spec_norm
conv_weights = {}
with tf.variable_scope(scope):
if zero:
conv_weights['c'] = get_weight(
'c', [k, k, c_in, c_out], spec_norm=spec_norm, zero=True)
else:
conv_weights['c'] = get_weight(
'c', [k, k, c_in, c_out], spec_norm=spec_norm)
conv_weights['b'] = tf.get_variable(
shape=[c_out], name='b', initializer=tf.initializers.zeros())
if classes != 1:
conv_weights['g'] = tf.get_variable(
shape=[
classes,
c_out],
name='g',
initializer=tf.initializers.ones())
conv_weights['gb'] = tf.get_variable(
shape=[
classes,
c_in],
name='gb',
initializer=tf.initializers.zeros())
else:
conv_weights['g'] = tf.get_variable(
shape=[c_out], name='g', initializer=tf.initializers.ones())
conv_weights['gb'] = tf.get_variable(
shape=[c_in], name='gb', initializer=tf.initializers.zeros())
conv_weights['cb'] = tf.get_variable(
shape=[c_in], name='cb', initializer=tf.initializers.zeros())
weights[scope] = conv_weights