我想在外部文件中写入所有标志及其值(如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
答案 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
,它不能显示可读的值。