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

Disable writing examples/sec and global_step/sec summaries on TPU if save_summary_steps is Falsey.

Browse Source 
PiperOrigin-RevId: 228373932
This commit is contained in:
Adam Roberts 2019-01-08 12:03:30 -08:00 committed by TensorFlower Gardener
parent 68727289bc
commit 989c8b40a9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2654,7 +2654,7 @@ class TPUEstimator(estimator_lib.Estimator):
if self._log_every_n_steps is not None:
examples_hook = ExamplesPerSecondHook(
ctx.global_batch_size,
output_dir=self.model_dir,
output_dir=self.model_dir if config.save_summary_steps else None,
every_n_steps=self._log_every_n_steps)
if ctx.is_running_on_cpu(is_export_mode=is_export_mode):