tf.estimator.SessionRunHook

Hook to extend calls to MonitoredSession.run().

Methods

after_create_session

View source

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.

Args
session A TensorFlow Session that has been created.
coord A Coordinator object which keeps track of all threads.

after_run

View source

Called after each call to run().

The run_values argument contains results of requested ops/tensors by before_run()</