我想在外部文件(如txt)中写入所有标志及其值。如何自动获取tf.flag
内的所有内容?有什么内置功能吗?或者有简单的方法,例如使用循环?
tf.flags.DEFINE_string("device","/gpu:0", "select device")
tf.flags.DEFINE_integer("rnn_size","64", "number of units")
我想得到
device /gpu:0
rnn_size 64
对于tensorflow 1.5,您可以使用tf.app.flags.FLAGS.flag_values_dict()
,他们再次更改了标志库
查看源代码,似乎API不直接支持它。如果你需要一个hack,你可以使用tf.flags.FLAGS.__flags
获取字典
完整代码如下:
特遣部队1.13# get flags
for attr,flag_obj in tf.FLAGS.__flags.items():
print("attr:%stvalue:%s" % (attr, flag_obj.value))
我看到没有人提到__flags
中key的值是flag object
,它不能打印可读值