diff options
Diffstat (limited to 'tensorflow/python/training/monitored_session.py')
| -rw-r--r-- | tensorflow/python/training/monitored_session.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/training/monitored_session.py b/tensorflow/python/training/monitored_session.py index fcec3ed97c..a891bae5f2 100644 --- a/tensorflow/python/training/monitored_session.py +++ b/tensorflow/python/training/monitored_session.py @@ -559,7 +559,7 @@ class MonitoredSession(_MonitoredSession): ```python saver_hook = CheckpointSaverHook(...) - summary_hook = SummaryHook(...) + summary_hook = SummarySaverHook(...) with MonitoredSession(session_creator=ChiefSessionCreator(...), hooks=[saver_hook, summary_hook]) as sess: while not sess.should_stop(): @@ -648,7 +648,7 @@ class SingularMonitoredSession(_MonitoredSession): Example usage: ```python saver_hook = CheckpointSaverHook(...) - summary_hook = SummaryHook(...) + summary_hook = SummarySaverHook(...) with SingularMonitoredSession(hooks=[saver_hook, summary_hook]) as sess: while not sess.should_stop(): sess.run(train_op) |