Class StepContext
Aliases:
- Class
tf.train.MonitoredSession.StepContext - Class
tf.train.SingularMonitoredSession.StepContext
Defined in tensorflow/python/training/monitored_session.py.
Control flow instrument for the step_fn from run_step_fn().
Users of step_fn may perform run() calls without running hooks
by accessing the session. A run() call with hooks may be performed
using run_with_hooks(). Computation flow can be interrupted using
request_stop().
__init__
__init__(
session,
run_with_hooks_fn
)
Initializes the step_context argument for a step_fn invocation.
Args:
session: An instance oftf.Session.run_with_hooks_fn: A function for running fetches and hooks.
Properties
session
Methods
tf.train.MonitoredSession.StepContext.request_stop
request_stop()
Exit the training loop by causing should_stop() to return True.
Causes step_fn to exit by raising an exception.
Raises:
StopIteration
tf.train.MonitoredSession.StepContext.run_with_hooks
run_with_hooks(
*args,
**kwargs
)
Same as MonitoredSession.run. Accepts the same arguments.