text tf.app.flags.FLAGS用法

Posted 2021-05-01

#tf.app.flags.FALGS是一个全局变量，用于定义命令行执行时可以传入的参数
FLAGS = tf.app.flags.FLAGS

tf.app.flags.DEFINE_string('train_dir', '/tmp/cifar10_train',
                           """Directory where to write event logs """
                           """and checkpoint.""")
tf.app.flags.DEFINE_integer('max_steps', 1000000,
                            """Number of batches to run.""")
tf.app.flags.DEFINE_boolean('log_device_placement', False,
                            """Whether to log device placement.""")
tf.app.flags.DEFINE_integer('log_frequency', 10,
                            """How often to log results to the console.""")
                            。
                            。
                            。
                            。
def main(unused_argv):
  train_dir =FLAGS.train_dir
  ........


if __name__=='__main__':
  tf.app.run()

#在通过命令行执行该文件的时候，可以python name.py train_dir="" max_steps=90 .....