Skip to content

GitLab

Commit 0b09300c authored by JimintheBox's avatar JimintheBox Committed by aquariusjay
Browse files

Fix vis.py for the issue Waiting for new Checkpoint (#6789)

parent 03027e33
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
research/deeplab/vis.py
View file @ 0b09300c
...@@ -276,10 +276,7 @@ def main(unused_argv): ...@@ -276,10 +276,7 @@ def main(unused_argv):
checkpoints_iterator = tf.contrib.training.checkpoints_iterator( checkpoints_iterator = tf.contrib.training.checkpoints_iterator(
FLAGS.checkpoint_dir, min_interval_secs=FLAGS.eval_interval_secs) FLAGS.checkpoint_dir, min_interval_secs=FLAGS.eval_interval_secs)
for checkpoint_path in checkpoints_iterator: for checkpoint_path in checkpoints_iterator:
if max_num_iteration > 0 and num_iteration > max_num_iteration:
break
num_iteration += 1 num_iteration += 1
tf.logging.info( tf.logging.info(
'Starting visualization at ' + time.strftime('%Y-%m-%d-%H:%M:%S', 'Starting visualization at ' + time.strftime('%Y-%m-%d-%H:%M:%S',
time.gmtime())) time.gmtime()))
...@@ -313,6 +310,8 @@ def main(unused_argv): ...@@ -313,6 +310,8 @@ def main(unused_argv):
tf.logging.info( tf.logging.info(
'Finished visualization at ' + time.strftime('%Y-%m-%d-%H:%M:%S', 'Finished visualization at ' + time.strftime('%Y-%m-%d-%H:%M:%S',
time.gmtime())) time.gmtime()))
if max_num_iteration > 0 and num_iteration >= max_num_iteration:
break
if __name__ == '__main__': if __name__ == '__main__':
flags.mark_flag_as_required('checkpoint_dir') flags.mark_flag_as_required('checkpoint_dir')
......
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