[TensorFlow] Flags

flags = tf.app.flags 이 함수는 python의 argparse 모듈과 같은 역할을 한다.

tf.app.flags로 flags 객체에 접근하여 DEFINE_* 함수에 (key, value, document)형식으로 설정하여 flags.FLAGS를 통해 어디서든 호출하여 사용 할 수 있다.

import tensorflow as tf
 
flags = tf.app.flags
FLAGS = flags.FLAGS
 
flags.DEFINE_float('learning_rate', 0.01,
                   'Initial learning rate.')
flags.DEFINE_integer('max_steps', 2000,
                     'Number of steps to run trainer.')
flags.DEFINE_integer('hidden1', 128,
                     'Number of units in hidden layer 1.')
flags.DEFINE_integer('hidden2', 32,
                     'Number of units in hidden layer 2.')
flags.DEFINE_integer('batch_size', 100,
                     'Batch size.'
                     'Must divide evenly into the dataset sizes.')
flags.DEFINE_string('train_dir', 'data',
                    'Directory to put the training data.')
flags.DEFINE_boolean('fake_data', False,
                     'If true, uses fake data '
                     'for unit testing.')
 
print(FLAGS.learning_rate)
print(FLAGS.batch_size)

>> 0.01

100