TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.train.StopAtStepHook

View source on GitHub

Class StopAtStepHook

Hook that requests stop at a specified step.

Inherits From: SessionRunHook

Aliases:

  • Class tf.compat.v1.estimator.StopAtStepHook
  • Class tf.compat.v1.train.StopAtStepHook
  • Class tf.compat.v2.estimator.StopAtStepHook
  • Class tf.estimator.StopAtStepHook

__init__

View source

__init__(
    num_steps=None,
    last_step=None
)

Initializes a StopAtStepHook.

This hook requests stop after either a number of steps have been executed or a last step has been reached. Only one of the two options can be specified.

if num_steps is specified, it indicates the number of steps to execute after begin() is called. If instead last_step is specified, it indicates the last step we want to execute, as passed to the after_run() call.

Args:

  • num_steps: Number of steps to execute.
  • last_step: Step after which to stop.

Raises:

  • ValueError: If one of the arguments is invalid.

Methods

after_create_session

View source

after_create_session(
    session,
    coord
)

after_run

View source

after_run(
    run_context,
    run_values
)

before_run

View source

before_run(run_context)

begin

View source

begin()

end

View source

end(session)

Called at the end of session.

The session argument can be used in case the hook wants to run final ops, such as saving a last checkpoint.

If session.run() raises exception other than OutOfRangeError or StopIteration then end() is not called. Note the difference between end() and after_run() behavior when session.run() raises OutOfRangeError or StopIteration. In that case end() is called but after_run() is not called.

Args:

  • session: A TensorFlow Session that will be soon closed.