Missed TensorFlow World? Check out the recap. Learn more

tf.estimator.experimental.make_stop_at_checkpoint_step_hook

View source on GitHub

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(
    estimator,
    last_step,
    wait_after_file_check_secs=30
)