Code 공부/tensorflow

[Tensorflow] 모듈공부 -01 tf.flags

디디52 2020. 2. 3. 15:31

<이글은 개인공부 기록을 위해 작성한 글입니다. 오류가 있을 수 있습니다.>

 

Flags 객체를 사용하면 고정값으로 되어 있는 기본적인 데이터를 편리하게 사용할 수 있다.

 

int, float, boolean, string 등의 값을 저장하고 가져다가 사용하기 쉽게 해주는 기능을 한다.

 

String 예시.

FLAGS = tf.flags.FLAGS

tf.flags.DEFINE_string('process', 'test', 'which process you want to do: write(write tfrecord file), train(train model), test(test model)')
tf.flags.DEFINE_string('imagedir', '../mnist/train_img', 'directory where images are stored in')
tf.flags.DEFINE_string('datadir', '../mnist/test_tfrecord', 'directory where tfrecord files are stored in')
tf.flags.DEFINE_string('val_datadir', '../mnist/val_tfrecord', 'directory where validation tfrecord files are stored in')
tf.flags.DEFINE_string('ckptdir', '../ckpt', 'checkpoint directory')