Internal change

PiperOrigin-RevId: 315383689

Internal change
PiperOrigin-RevId: 315383689
4c77f5b1 · Hongkun Yu · A. Unique TensorFlower · e82a2d9f · 4c77f5b1
Commit 4c77f5b1 authored Jun 08, 2020 by Hongkun Yu Committed by A. Unique TensorFlower Jun 08, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

official/utils/misc/keras_utils.py official/utils/misc/keras_utils.py +2 -2

No files found.
--- a/official/utils/misc/keras_utils.py
+++ b/official/utils/misc/keras_utils.py
@@ -133,9 +133,9 @@ class TimeHistory(tf.keras.callbacks.Callback):

      if self.summary_writer:
        with self.summary_writer.as_default():
-          tf.summary.scalar('global_step/sec', steps_per_second,
+          tf.summary.scalar('steps_per_second', steps_per_second,
                            self.global_steps)
-          tf.summary.scalar('examples/sec', examples_per_second,
+          tf.summary.scalar('examples_per_second', examples_per_second,
                            self.global_steps)

      self.last_log_step = self.global_steps