Merge pull request #1732 from mozilla/issue1505

Disable StepCounterHook to avoid useless warning during validation (Fixes #1505)
This commit is contained in:
Reuben Morais 2018-11-17 16:07:34 -02:00 committed by GitHub
commit c33b0d29d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -540,6 +540,7 @@ def train(server=None):
hooks=hooks,
checkpoint_dir=FLAGS.checkpoint_dir,
save_checkpoint_secs=None, # already taken care of by a hook
log_step_count_steps=0, # disable logging of steps/s to avoid TF warning in validation sets
config=Config.session_config) as session:
tf.get_default_graph().finalize()