#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 .....