experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Disable CheckpointSaverHook when both save_checkpoints_secs and

Browse Source 
save_checkpoints_steps are None

PiperOrigin-RevId: 193299688
This commit is contained in:
Chris Ying 2018-04-17 20:31:30 -07:00 committed by TensorFlower Gardener
parent 235a9e32c5
commit 41e2cd187b
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tensorflow/contrib/tpu/python/tpu/tpu_estimator.py
View File

@ -2054,14 +2054,16 @@ class TPUEstimator(estimator_lib.Estimator):
},
every_n_secs=30)
] + input_hooks
chief_hooks = [
training.CheckpointSaverHook(
self.model_dir,
save_secs=self._config.save_checkpoints_secs,
save_steps=self._config.save_checkpoints_steps,
steps_per_run=self._config.tpu_config.iterations_per_loop,
scaffold=scaffold)
]
chief_hooks = []
if (self._config.save_checkpoints_secs or
self._config.save_checkpoints_steps):
chief_hooks.append(
training.CheckpointSaverHook(
self.model_dir,
save_secs=self._config.save_checkpoints_secs,
save_steps=self._config.save_checkpoints_steps,
steps_per_run=self._config.tpu_config.iterations_per_loop,
scaffold=scaffold))
summary.scalar(model_fn_lib.LOSS_METRIC_KEY, loss)
with ops.control_dependencies([loss]):
update_ops = _sync_variables_ops()