如何在记录器中写张量流标志变量

时间:2019-06-24 16:05:16

标签: python tensorflow argparse flags

我正在使用tensowflow标志来解析如下参数:

flags = tf.app.flags
FLAGS = flags.FLAGS
lags.DEFINE_float('learning_rate', 0.01, 'Initial learning rate.')
flags.DEFINE_integer('epochs', 200, 'Number of epochs to train.')
flags.DEFINE_integer('batch_size', 32, 'Number of graphs in a batch.')
flags.DEFINE_integer('hidden1', 5, 'Number of units in hidden layer 1.')

我想将这些参数写入记录器。我知道在此处使用“ argparse”而不是标志时,我们可以使用以下代码将其写入记录器:

logger.info(vars(args)); pprint(vars(args))

但是当我使用

logger.info(vars(FLAGS))

它显示

{'__wrapped': <absl.flags._flagvalues.FlagValues object at 0x11dfc1da0>}
记录器中的

,而不是特定参数。那么如何处理“标志”呢?

1 个答案:

答案 0 :(得分:0)

我发现您只需删除varslogger.info(FLAGS)就可以给出参数的值。