Merge pull request #1318 from mozilla/issue1317

Disable checkpoint saving in MonitoredTrainingSession (Fixes #1317)
This commit is contained in:
Reuben Morais 2018-03-27 09:50:38 -03:00 committed by GitHub
commit 31c01db541
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1579,7 +1579,7 @@ def train(server=None):
is_chief=is_chief,
hooks=hooks,
checkpoint_dir=FLAGS.checkpoint_dir,
save_checkpoint_secs=FLAGS.checkpoint_secs if FLAGS.train else None,
save_checkpoint_secs=None, # already taken care of by a hook
config=session_config) as session:
if len(FLAGS.initialize_from_frozen_model) > 0:
log_info('Initializing from frozen model: {}'.format(FLAGS.initialize_from_frozen_model))