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
)