TensorFlow BN更新

2018-07-28 本文已影响0人翻开日记

update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS)
update_ops.append(train_op)
update_op = tf.group(*update_ops)