如何使用Tensorflow中的loop访问所有标志并获取它们的值?

2024-06-07 03:59:03 发布

您现在位置:Python中文网/ 问答频道 /正文

我想在外部文件(如txt)中写入所有标志及其值。 如何自动获取tf.flag中的所有内容? 有内置功能吗?还是有简单的方法,例如使用loop?

例如

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

Tags: 文件方法功能txt内容sizegpu标志
2条回答

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

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

相关问题 更多 >