tf.estimator.experimental.make_stop_at_checkpoint_step_hook

View source on GitHub

Creates a proper StopAtCheckpointStepHook based on chief status.

tf.estimator.experimental.make_stop_at_checkpoint_step_hook(
    estimator, last_step, wait_after_file_check_secs=30
)

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/estimator/experimental/make_stop_at_checkpoint_step_hook