如何在Tensorflow中使用循环访问所有标志并获取它们的值



我想在外部文件(如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,它不能打印可读值

最新更新