У меня есть базовый 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