Tensorflow:tf.app.flags.DEFINE_string()和tf.app.flags.FLAGS使用

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/FUZHENQI/article/details/80857767

tf.app.flags.DEFINE_xxx():添加命令行的optional argument(可选参数)。
而tf.app.flags.FLAGS:从对应的命令行参数取出参数。
举个栗子
新建test.py文件,并输入如下代码,代码的功能是创建几个命令行参数,然后把命令行参数输出显示

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_string(
    'flag_string', 'yes', '')

print(FLAGS.flag_float)
print(FLAGS.flag_int)
print(FLAGS.flag_string)

1.在命令行中查看帮助信息

 python test.py -h

这里写图片描述
2.直接运行test.py

python test.py

输出默认值:
这里写图片描述
3.带命令行参数的运行test.py文件

python test.py --flag_int 888 --flag_float 0.222 --flag_string 'I love Python !'

输出参数值:
这里写图片描述
注:参数类型详见https://blog.csdn.net/m0_37041325/article/details/77448971

猜你喜欢

转载自blog.csdn.net/FUZHENQI/article/details/80857767