diff options
author | 2017-03-31 09:42:41 -0800 | |
---|---|---|
committer | 2017-03-31 11:06:27 -0700 | |
commit | 997ffb515f33911bde527465ab886a93a3cf9e67 (patch) | |
tree | ceb6a0b22ddcf68d2be43e57f83b1dccbaa2b03a /tensorflow/python/training/monitored_session.py | |
parent | 0fe523ac05f1f49145f4b243953a7aad331ea4dc (diff) |
Add default saver option to CheckpointSaverHook and improve docstrings.
Change: 151839983
Diffstat (limited to 'tensorflow/python/training/monitored_session.py')
-rw-r--r-- | tensorflow/python/training/monitored_session.py | 7 |
1 files changed, 1 insertions, 6 deletions
diff --git a/tensorflow/python/training/monitored_session.py b/tensorflow/python/training/monitored_session.py index ae76a1ab58..cf8692eda1 100644 --- a/tensorflow/python/training/monitored_session.py +++ b/tensorflow/python/training/monitored_session.py @@ -22,7 +22,6 @@ from __future__ import print_function import abc from tensorflow.core.protobuf import config_pb2 -from tensorflow.core.protobuf import saver_pb2 from tensorflow.python.framework import errors from tensorflow.python.framework import ops from tensorflow.python.ops import array_ops @@ -180,11 +179,7 @@ class Scaffold(object): summary.merge_all) # pylint: disable=g-long-lambda if self._saver is None: - self._saver = Scaffold.get_or_default( - 'saver', - ops.GraphKeys.SAVERS, - lambda: training_saver.Saver(sharded=True, allow_empty=True, - write_version=saver_pb2.SaverDef.V2)) + self._saver = training_saver._get_saver_or_default() # pylint: disable=protected-access # pylint: enable=g-long-lambda self._saver.build() |