tf.flags是用来设置参数的,以便后面的程序用到,感觉跟变量的作用是一样的。
# coding: utf-8
import tensorflow as tf
# define flags
tf.flags.DEFINE_integer("age", 17, "age of user(default:20)")
tf.flags.DEFINE_boolean("drink_allow", False, "if can drink or not(default:False)")
tf.flags.DEFINE_float("weight", 55.55, "weight of user(default:55.55kg)")
tf.flags.DEFINE_string("name", "Lilei", "name of user(default:Lilei)")
FLAGS = tf.flags.FLAGS #init flags
FLAGS._parse_flags() # parse flags
# get flags
for attr,value in FLAGS.__flags.items():
print("attr:%s\tvalue:%s" % (attr,str(value)))
attr:name value:Lilei
attr:weight value:55.55
attr:age value:17
attr:drink_allow value:False
FLAGS.name = 'zhangshan'
# get flags
for attr,value in FLAGS.__flags.items():
print("attr:%s\tvalue:%s" % (attr,str(value)))
attr:name value:zhangshan
attr:drink_allow value:False
attr:age value:17
attr:weight value:55.55
本文介绍了如何使用TensorFlow中的tf.flags来定义和获取程序参数,包括整型、布尔型、浮点型和字符串类型的参数,并展示了如何修改这些参数的值。
4705

被折叠的 条评论
为什么被折叠?



