tf.train.StepCounterHook
tf.train.StepCounterHook
class tf.train.StepCounterHook
Defined in tensorflow/python/training/basic_session_run_hooks.py
.
See the guide: Training > Training Hooks
Hook that counts steps per second.
Methods
__init__
__init__( every_n_steps=100, every_n_secs=None, output_dir=None, summary_writer=None )
after_create_session
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.
Args:
-
session
: A TensorFlow Session that has been created. -
coord
: A Coordinator object which keeps track of all threads.
after_run
after_run( run_context, run_values )
before_run
before_run(run_context)
begin
begin()
end
end(session)
Called at the end of session.
The session
argument can be used in case the hook wants to run final ops, such as saving a last checkpoint.
If session.run()
raises exception other than OutOfRangeError or StopIteration then end()
is not called. Note the difference between end()
and after_run()
behavior when session.run()
raises OutOfRangeError or StopIteration. In that case end()
is called but after_run()
is not called.
Args:
-
session
: A TensorFlow Session that will be soon closed.
© 2017 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/train/StepCounterHook