Получение показателей эффективности обучения из Tensorflow Estimator - PullRequest
1 голос
/ 04 октября 2019

У меня есть базовый tst TPUEstimator, который я использую для обучения своей нейронной сети, он успешно обучается и регистрирует убывающие потери как в журнале stdout, так и в выходных файлах журнала TensorBoard.

Однако я не могунайдите метрики train_examples/sec или train_global_step/sec. У меня установлено tf.logging.set_verbosity(tf.logging.INFO), но эти метрики не отображаются в консоли или на Tensorboard.

Есть ли какой-либо LoggingHook, который мне нужно добавить при обучении? Альтернативно, есть ли другой способ измерения шагов или примеров в секунду как метрики?

Версия Tensorflow - 1.14 и Python3.7

1 Ответ

0 голосов
/ 11 октября 2019

Вы устанавливаете очень высокое значение (или 0) для log_step_count_steps в tf.estimator.RunConfig ? Кроме того, вы можете вручную добавить tf.train.StepCounterHook .

...