如何在记录器中写入 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" 而不是 flags 时，我们可以使用下面的代码写入 logger：

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

但是当我使用

logger.info(vars(FLAGS))

显示

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

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

Answer 1

我发现你直接drop vars，logger.info(FLAGS)就可以给出参数的值

如何在记录器中写入 tensorflow 标志变量

How to write tensorflow flag variables in a logger

python

flags

argparse

tensorflow