Skip to content

Commit

Permalink
update cnn base net according to tf1.10 new api
Browse files Browse the repository at this point in the history
  • Loading branch information
MaybeShewill-CV committed Oct 25, 2018
1 parent 05a699e commit bcff44a
Showing 1 changed file with 0 additions and 23 deletions.
23 changes: 0 additions & 23 deletions encoder_decoder_model/cnn_basenet.py
Original file line number Diff line number Diff line change
Expand Up @@ -312,29 +312,6 @@ def layerbn(inputdata, is_training, name):
:param name:
:return:
"""
# def f1():
# """
#
# :return:
# """
# # print('batch_normalization: train phase')
# return tf_layer.batch_norm(
# inputdata, is_training=True,
# center=True, updates_collections=None,
# scope=name, reuse=False)
#
# def f2():
# """
#
# :return:
# """
# # print('batch_normalization: test phase')
# return tf_layer.batch_norm(
# inputdata, is_training=False,
# center=True, updates_collections=None,
# scope=name, reuse=True)
#
# output = tf.cond(is_training, f1, f2)

return tf.layers.batch_normalization(inputs=inputdata, training=is_training, name=name)

Expand Down

0 comments on commit bcff44a

Please sign in to comment.