TensorFlow 2.0 Beta is available Learn more

tf.estimator.experimental.make_stop_at_checkpoint_step_hook

Creates a proper StopAtCheckpointStepHook based on chief status.

Aliases:

  • tf.compat.v1.estimator.experimental.make_stop_at_checkpoint_step_hook
  • tf.compat.v2.estimator.experimental.make_stop_at_checkpoint_step_hook
  • tf.estimator.experimental.make_stop_at_checkpoint_step_hook
tf.estimator.experimental.make_stop_at_checkpoint_step_hook(
    estimator,
    last_step,
    wait_after_file_check_secs=30
)

Defined in python/estimator/hooks/hooks.py.