tensorflow 使用flags定义命令行参数的方法
tf定义了tf.app.flags，用于支持接受命令行传递参数，相当于接受argv。
importtensorflowastf
#第一个是参数名称，第二个参数是默认值，第三个是参数描述
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()takesnoarguments(1given)';main的参数名随意定义，无要求
defmain(_):
print(FLAGS.str_name)
print(FLAGS.int_name)
print(FLAGS.bool_name)
if__name__=='__main__':
tf.app.run()#执行main函数
执行：
[root@AliHPC-G41-211test]#pythontt.py
def_v_1
10
False
[root@AliHPC-G41-211test]#pythontt.py--str_nametest_str--int_name99--bool_nameTrue
test_str
99
True
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。