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

 

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')

'Code 공부 > tensorflow' 카테고리의 다른 글

[Tensorflow] 02 - Tf.Record  (0) 2020.02.04

+ Recent posts