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

Minor improved log statements for CheckpointSaverHook.

Browse Source 
PiperOrigin-RevId: 292692483
Change-Id: I07ab95f6e99e45f45e4cadba28926d4ebe44cb53
This commit is contained in:
A. Unique TensorFlower 2020-02-01 03:20:55 -08:00 committed by TensorFlower Gardener
parent 24fd250d73
commit 073460df14
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/training/basic_session_run_hooks.py
View File

@ -610,18 +610,20 @@ class CheckpointSaverHook(session_run_hook.SessionRunHook):
def _save(self, session, step): def _save(self, session, step):
"""Saves the latest checkpoint, returns should_stop.""" """Saves the latest checkpoint, returns should_stop."""
logging.info("Saving checkpoints for %d into %s.", step, self._save_path) logging.info("Calling checkpoint listeners before saving checkpoint %d...",
step)
for l in self._listeners: for l in self._listeners:
l.before_save(session, step) l.before_save(session, step)
logging.info("Saving checkpoints for %d into %s.", step, self._save_path)
self._get_saver().save(session, self._save_path, global_step=step, self._get_saver().save(session, self._save_path, global_step=step,
write_meta_graph=self._save_graph_def) write_meta_graph=self._save_graph_def)
self._summary_writer.add_session_log( self._summary_writer.add_session_log(
SessionLog( SessionLog(
status=SessionLog.CHECKPOINT, checkpoint_path=self._save_path), status=SessionLog.CHECKPOINT, checkpoint_path=self._save_path),
step) step)
logging.info("Calling checkpoint listeners after saving checkpoint %d...",
step)
should_stop = False should_stop = False
for l in self._listeners: for l in self._listeners:
if l.after_save(session, step): if l.after_save(session, step):