Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.estimator.experimental.make_stop_at_checkpoint_step_hook

TensorFlow 2.0 version 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
)