如何在记录器中编写TensorFlow标志变量

Question

我正在使用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：

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

但是当我使用

时

logger.info(vars(FLAGS))

它显示

{'__wrapped': <absl.flags._flagvalues.FlagValues object at 0x11dfc1da0>}

在记录器中，而不是特定参数。那么如何处理"标志"？

Answer 1

我发现您可以删除vars，logger.info(FLAGS)可以给出参数的值。