TensorFlow之命令行参数

  TensorFlow定义了tf.app.flags(也可以用它的别名tf.flags)，用于支持接受命令行传递参数，相当于接收argv：

import tensorflow as tf

# 第一个是参数名称，第二个参数是默认值，第三个是参数描述
tf.app.flags.DEFINE_string('str_name', 'def_v_1', "descript1")
tf.app.flags.DEFINE_integer('int_name', 10, "descript2")
tf.app.flags.DEFINE_boolean('bool_name', False, "descript3")
FLAGS = tf.app.flags.FLAGS

def main(_):
    print(FLAGS.str_name)
    print(FLAGS.int_name)
    print(FLAGS.bool_name)

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

执行结果：

$ python test.py --str_name "hello" --int_name 12 --bool_name True
hello
12
True