Skip to content

GitLab

Commit ea550ca9 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

Merge branch 'master' of https://github.com/tensorflow/models into context_tf2

parents 3ce2f61b 6794c6b3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
orbit/controller.py
View file @ ea550ca9
...@@ -341,8 +341,8 @@ class Controller(object): ...@@ -341,8 +341,8 @@ class Controller(object):
# Calculates steps to run for the next train loop. # Calculates steps to run for the next train loop.
current_step = self.global_step.numpy() current_step = self.global_step.numpy()
logging.info("Entering training loop at step %s of %s", current_step, logging.info("Entering training loop at step %s to run %s steps",
num_steps) current_step, num_steps)
current_step += num_steps current_step += num_steps
num_steps = tf.convert_to_tensor(num_steps, dtype=tf.int32) num_steps = tf.convert_to_tensor(num_steps, dtype=tf.int32)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment