我正在使用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>}
记录器中的,而不是特定参数。那么如何处理“标志”呢?
答案 0 :(得分:0)
我发现您只需删除vars
,logger.info(FLAGS)
就可以给出参数的值。