Command-line-interface debugger hook.
Can be used as a hook for
tf.estimator.Estimators. Provides a substitute for
tfdbg.LocalCLIDebugWrapperSession in cases where the session is not directly
__init__( ui_type='curses', dump_root=None, thread_name_filter=None )
Create a local debugger command-line interface (CLI) hook.
str) requested user-interface type. Currently supported: (curses | readline).
str) optional path to the dump root directory. Must be a directory that does not exist or an empty directory. If the directory does not exist, it will be created by the debugger core during debug
run()calls and removed afterwards.
thread_name_filter: Regular-expression white list for threads on which the wrapper session will be active. See doc of
BaseDebugWrapperSessionfor more details.
add_tensor_filter( filter_name, tensor_filter )
Add a tensor filter.
See doc of
LocalCLIDebugWrapperSession.add_tensor_filter() for details.
Override default behavior to accommodate the possibility of this method
called prior to the initialization of the underlying
filter_name: See doc of
tensor_filter: See doc of
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.
session: A TensorFlow Session that has been created.
coord: A Coordinator object which keeps track of all threads.
after_run( run_context, run_values )
Called at the end of session.
session argument can be used in case the hook wants to run final ops,
such as saving a last checkpoint.
session.run() raises exception other than OutOfRangeError or
end() is not called.
Note the difference between
after_run() behavior when
session.run() raises OutOfRangeError or StopIteration. In that case
end() is called but
after_run() is not called.
session: A TensorFlow Session that will be soon closed.