如何使用Tensorflow中的循环访问所有标志并获取其值?

时间:2016-10-27 13:47:19

标签: python 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

3 个答案:

答案 0 :(得分:16)

对于tensorflow 1.5,您可以使用tf.app.flags.FLAGS.flag_values_dict()他们再次更改了标记库

答案 1 :(得分:8)

查看the source,API似乎不直接支持它。如果您需要hack,可以使用tf.flags.FLAGS.__flags来获取字典。

答案 2 :(得分:0)

完整的代码如下所示:

tf 1.13

# get flags
for attr,flag_obj in tf.FLAGS.__flags.items():
    print("attr:%s\tvalue:%s" % (attr, flag_obj.value))

我发现没有人提及__flags中的key的值为flag object,它不能显示可读的值。