tf.train.StopAtStepHook

Class StopAtStepHook

Inherits From: SessionRunHook

Defined in tensorflow/python/training/basic_session_run_hooks.py.

See the guide: Training > Training Hooks

Hook that requests stop at a specified step.

Methods

__init__

__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.

after_create_session

after_create_session(
    session,
    coord
)

after_run

after_run(
    run_context,
    run_values
)

before_run

before_run(run_context)

begin

begin()

end

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.