Improve _train_n_steps logging message.

PiperOrigin-RevId: 320855556

Improve _train_n_steps logging message.
PiperOrigin-RevId: 320855556
19dec974 · Hongkun Yu · A. Unique TensorFlower · efa796f8 · 19dec974
Commit 19dec974 authored Jul 12, 2020 by Hongkun Yu Committed by A. Unique TensorFlower Jul 12, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

orbit/controller.py orbit/controller.py +2 -2

No files found.
--- a/orbit/controller.py
+++ b/orbit/controller.py
@@ -341,8 +341,8 @@ class Controller(object):

    # Calculates steps to run for the next train loop.
    current_step = self.global_step.numpy()
-    logging.info("Entering training loop at step %s of %s", current_step,
-                 num_steps)
+    logging.info("Entering training loop at step %s to run %s steps",
+                 current_step, num_steps)
    current_step += num_steps
    num_steps = tf.convert_to_tensor(num_steps, dtype=tf.int32)