From 62915c0c528c8829502d8fa550c7c248a924c094 Mon Sep 17 00:00:00 2001 From: Mustafa Ispir Date: Wed, 18 Jan 2017 12:49:54 -0800 Subject: Added duration to the log of LoggingTensorHook. Duration is logged in many internal examples. Change: 144869279 --- tensorflow/python/training/basic_session_run_hooks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/training/basic_session_run_hooks.py b/tensorflow/python/training/basic_session_run_hooks.py index 206cfa442f..cda4fedec7 100644 --- a/tensorflow/python/training/basic_session_run_hooks.py +++ b/tensorflow/python/training/basic_session_run_hooks.py @@ -173,14 +173,14 @@ class LoggingTensorHook(session_run_hook.SessionRunHook): if self._should_trigger: original = np.get_printoptions() np.set_printoptions(suppress=True) + elapsed_secs, _ = self._timer.update_last_triggered_step(self._iter_count) if self._formatter: logging.info(self._formatter(run_values.results)) else: stats = [] for tag in self._tag_order: stats.append("%s = %s" % (tag, run_values.results[tag])) - logging.info("%s", ", ".join(stats)) - self._timer.update_last_triggered_step(self._iter_count) + logging.info("%s (%.3f sec)", ", ".join(stats), elapsed_secs) np.set_printoptions(**original) self._iter_count += 1 -- cgit v1.2.3