View source on GitHub
|
Control flow instrument for the step_fn from run_step_fn().
tf.compat.v1.train.MonitoredSession.StepContext(
session, run_with_hooks_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().
Args | |
|---|---|
session
|
An instance of tf.compat.v1.Session.
|
run_with_hooks_fn
|
A function for running fetches and hooks. |
Attributes | |
|---|---|
session
|
|
Methods
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 |
run_with_hooks
run_with_hooks(
*args, **kwargs
)
Same as MonitoredSession.run. Accepts the same arguments.
View source on GitHub