diff options
author | Sourabh Bajaj <sourabhbajaj@google.com> | 2018-08-10 21:30:58 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-10 21:35:14 -0700 |
commit | 054b3961f76a12bc8b722af708982949b4e951cf (patch) | |
tree | 627ed5c7cddd9126b49a1d417d24cffef6495134 /tensorflow/python/estimator/estimator_test.py | |
parent | 84ee0e2a2554e9b9ccfbaf0db1e2db62dd52d8cc (diff) |
Create new MultiStepStopHook for running mutiple steps per run when using TPU Distribution Strategy with Estimator.
PiperOrigin-RevId: 208310915
Diffstat (limited to 'tensorflow/python/estimator/estimator_test.py')
-rw-r--r-- | tensorflow/python/estimator/estimator_test.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/estimator/estimator_test.py b/tensorflow/python/estimator/estimator_test.py index d44774a45a..e3f22d9010 100644 --- a/tensorflow/python/estimator/estimator_test.py +++ b/tensorflow/python/estimator/estimator_test.py @@ -2683,6 +2683,7 @@ class EstimatorExportTest(test.TestCase): _, _ = features, labels my_int = variables.Variable(1, name='my_int', collections=[ops.GraphKeys.LOCAL_VARIABLES]) + _ = training.get_or_create_steps_per_run_variable() scores = constant_op.constant([3.]) with ops.control_dependencies([ variables.local_variables_initializer(), |