【tensorflow】:tf.app.run() & tf.app.flags()

代码段


#_*_coding:utf-8_*_
import tensorflow as tf

# tensorflow定义了 tf.app.flags用来接收命令行带来的参数
flag = tf.app.flags

#param1: 变量名称
#param2:变量默认值
#param3:变量描述
flag.DEFINE_string(flag_name='name',default_value='yuanlei',docstring='username')
flag.DEFINE_integer(flag_name='age',default_value=24,docstring="'user's age")
flag.DEFINE_boolean(flag_name='sex',default_value=True,docstring='True:boy, False:girl')

Flags = flag.FLAGS

def main(temp):
    print(Flags.name)
    print(Flags.age)
    print(Flags.sex)

if __name__ == "__main__":
    #这个函数,就是首先处理flag解析,
    #再之后执行main()函数
    tf.app.run()

直接在pycharm中运行py文件输出结果(不传递参数的默认结果)

这里写图片描述

在terminal中采集参数输出结果

这里写图片描述

猜你喜欢

转载自blog.csdn.net/qiu931110/article/details/80111893