in utils.py [0:0]
def normalize(inp, activation, reuse, scope):
if FLAGS.norm == 'batch_norm':
return tf_layers.batch_norm(
inp,
activation_fn=activation,
reuse=reuse,
scope=scope)
elif FLAGS.norm == 'layer_norm':
return tf_layers.layer_norm(
inp,
activation_fn=activation,
reuse=reuse,
scope=scope)
elif FLAGS.norm == 'None':
if activation is not None:
return activation(inp)
else:
return inp