tf.app.flags.FLAGS 和 tf.app.flags.DEFINE

mac2022-06-30  174

tf.app.flags.DEFINE_xxx() 就是添加命令行的 optional argument(可选参数) 而tf.app.flags.FLAGS可以从对应的命令行参数取出参数

例如:

import tensorflow as tf FLAGS=tf.app.flags.FLAGS tf.app.flags.DEFINE_float( 'flag_float', 0.01, 'input a float') tf.app.flags.DEFINE_integer( 'flag_int', 400, 'input a int') tf.app.flags.DEFINE_boolean( 'flag_bool', True, 'input a bool') tf.app.flags.DEFINE_string( 'flag_string', 'yes', 'input a string') print(FLAGS.flag_float) print(FLAGS.flag_int) print(FLAGS.flag_bool) print(FLAGS.flag_string)

(1)在命令行中查看帮助信息,在命令行输入 python test.py -h 注意红色框中的信息,这个就是我们用DEFINE_XXX添加命令行参数时的第三个参数

(2)直接运行python test.py 因为没有给对应的命令行参数赋值,所以输出的是命令行参数的默认值 (3)带命令行参数的运行test.py文件

python test.py --flag_int 1 --flag_bool True --flag_string 'haha,I get it'-- flag_float 0.2

这里输出了我们赋给命令行参数的值

南淮北安 认证博客专家 深度学习 Java Python 牛年大吉,牛气冲天!!!
最新回复(0)