Tensorflow使用flags定义命令行参数详解

Never-Giveup

于 2018-07-19 23:29:29 发布

阅读量7.2k

点赞数 3

分类专栏： tensorflow函数用法随笔文章标签： Tensorflow flags 命令行参数解析 python

本文链接：https://blog.csdn.net/qq_36653505/article/details/81124533

版权

本文详述了TensorFlow中使用tf.app.flags定义并解析命令行参数的方法，包括通过示例一、二、三展示如何设置和修改默认参数值。通过tf.app.run()运行程序并调用_f.parse_flags()来解析参数，执行main函数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

TensorFlow定义了tf.app.flags，用于支持接受命令行传递参数，相当于接受argv，详细用法请看代码中的注释

例一：

import tensorflow as tf

##第一个是参数名称，第二个参数是默认值，第三个是参数描述
tf.app.flags.DEFINE_string('str_name', 'def_v_1',"descrip1")
tf.app.flags.DEFINE_integer('int_name', 10,"descript2")
tf.app.flags.DEFINE_boolean('bool_name', False, "descript3")
FLAGS = tf.app.flags.FLAGS
##必须带参数，否则：'TypeError: main() takes no arguments (1 given)';   ##main的参数名随意定义，无要求
def main(_):  
    print(FLAGS.str_name)
    print(FLAGS.int_name)
    print(FLAGS.bool_name)

if __name__ == '__main__':
    tf.app.run()  #tf.app.run()的作用：先处理fla

最低0.47元/天解锁文章