diff options
author | Mustafa Ispir <ispir@google.com> | 2018-08-16 14:45:43 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-16 14:56:33 -0700 |
commit | 1c7cafc092cf489b77f338478be5bde1721c5d71 (patch) | |
tree | 618393bb82adf2cf126f3bbd98ddd8954d0c48de /tensorflow/contrib/estimator/__init__.py | |
parent | 2a08dd7f020138f5b79af188504937321f4d542d (diff) |
Added a factory for StopAtCheckpointStepHook. Chief is responsible to save the checkpoint. Therefore StopAtCheckpointStepHook should not be used in chief mode.
PiperOrigin-RevId: 209051351
Diffstat (limited to 'tensorflow/contrib/estimator/__init__.py')
-rw-r--r-- | tensorflow/contrib/estimator/__init__.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/estimator/__init__.py b/tensorflow/contrib/estimator/__init__.py index 6ad3a4a604..258860f263 100644 --- a/tensorflow/contrib/estimator/__init__.py +++ b/tensorflow/contrib/estimator/__init__.py @@ -45,7 +45,7 @@ _allowed_symbols = [ 'clip_gradients_by_norm', 'forward_features', 'InMemoryEvaluatorHook', - 'StopAtCheckpointStepHook', + 'make_stop_at_checkpoint_step_hook', 'logistic_regression_head', 'multi_class_head', 'multi_head', |