tf.train.LoggingTensorHook
tf.train.LoggingTensorHook
class tf.train.LoggingTensorHook
Defined in tensorflow/python/training/basic_session_run_hooks.py
.
See the guide: Training > Training Hooks
Prints the given tensors once every N local steps or once every N seconds.
The tensors will be printed to the log, with INFO
severity.
Methods
__init__
__init__( tensors, every_n_iter=None, every_n_secs=None, formatter=None )
Initializes a LoggingTensorHook
.
Args:
-
tensors
:dict
that maps string-valued tags to tensors/tensor names, oriterable
of tensors/tensor names. -
every_n_iter
:int
, print the values oftensors
once every N local steps taken on the current worker. -
every_n_secs
:int
orfloat
, print the values oftensors
once every N seconds. Exactly one ofevery_n_iter
andevery_n_secs
should be provided. -
formatter
: function, takes dict oftag
->Tensor
and returns a string. IfNone
uses default printing all tensors.
Raises:
-
ValueError
: ifevery_n_iter
is non-positive.
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/LoggingTensorHook