|View source on GitHub|
Hook that requests stop at a specified step.
Compat aliases for migration
See Migration guide for more details.
tf.estimator.StopAtStepHook( num_steps=None, last_step=None )
||Number of steps to execute.|
||Step after which to stop.|
||If one of the arguments is invalid.|
after_create_session( session, coord )
Called when new TensorFlow session is created.
This is called to signal the hooks that a new session has been created. This
has two essential differences with the situation in which
begin is called:
- When this is called, the graph is finalized and ops can no longer be added to the graph.
- This method will also be called as a result of recovering a wrapped session, not only at the beginning of the overall session.
||A TensorFlow Session that has been created.|
||A Coordinator object which keeps track of all threads.|
after_run( run_context, run_values )
Called after each call to run().
run_values argument contains results of requested ops/tensors by
run_context argument is the same one send to
run_context.request_stop() can be called to stop the iteration.
session.run() raises any exceptions then
after_run() is not called.
||A SessionRunValues object.|
before_run( run_context )
Called before each call to run().
You can return from this call a