Как записать переменные флаг тензорного потока в логгер - PullRequest
0 голосов
/ 24 июня 2019

Я использую флаги tenowflow для разбора аргументов, таких как:

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 голосов
/ 26 июня 2019

Я обнаружил, что вы можете просто сбросить vars, logger.info(FLAGS) может дать значение параметров.

...