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/r2.4/api_docs/python/tf/estimator/experimental/make_stop_at_checkpoint_step_hook