diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-11 11:02:48 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-11 11:05:19 -0700 |
| commit | 5757d091a5c915b5ca99da7bc44feebdb374c569 (patch) | |
| tree | 7843ba2cbd75f5a24dd81a81fb1414d64fad4b5b /tensorflow/contrib/training | |
| parent | 8f753859dd50a4c8d25b99a7b57c61e0e5c20578 (diff) | |
Use tf.train.get_or_create_global_step() instead of deprecated variables.get_or_create_global_step().
PiperOrigin-RevId: 192476077
Diffstat (limited to 'tensorflow/contrib/training')
| -rw-r--r-- | tensorflow/contrib/training/python/training/evaluation.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/contrib/training/python/training/evaluation.py b/tensorflow/contrib/training/python/training/evaluation.py index 4bb53e8678..f7fd66d33f 100644 --- a/tensorflow/contrib/training/python/training/evaluation.py +++ b/tensorflow/contrib/training/python/training/evaluation.py @@ -138,7 +138,6 @@ from __future__ import print_function import time -from tensorflow.contrib.framework.python.ops import variables from tensorflow.python.ops import state_ops from tensorflow.python.platform import tf_logging as logging from tensorflow.python.summary import summary @@ -298,7 +297,7 @@ class SummaryAtEndHook(session_run_hook.SessionRunHook): def begin(self): if self._replace_summary_op: self._summary_op = summary.merge_all() - self._global_step = variables.get_or_create_global_step() + self._global_step = training_util.get_or_create_global_step() def after_create_session(self, session, coord): if self._summary_writer is None and self._log_dir: |