TensorFlow BN更新

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

你可能感兴趣的:(TensorFlow BN更新)