tf.train.SummarySaverHook
tf.train.SummarySaverHook
class tf.train.SummarySaverHook
Defined in tensorflow/python/training/basic_session_run_hooks.py
.
See the guide: Training > Training Hooks
Saves summaries every N steps.
Methods
__init__
__init__( save_steps=None, save_secs=None, output_dir=None, summary_writer=None, scaffold=None, summary_op=None )
Initializes a SummarySaverHook
.
Args:
-
save_steps
:int
, save summaries every N steps. Exactly one ofsave_secs
andsave_steps
should be set. -
save_secs
:int
, save summaries every N seconds. -
output_dir
:string
, the directory to save the summaries to. Only used if nosummary_writer
is supplied. -
summary_writer
:SummaryWriter
. IfNone
and anoutput_dir
was passed, one will be created accordingly. -
scaffold
:Scaffold
to get summary_op if it's not provided. -
summary_op
:Tensor
of typestring
containing the serializedSummary
protocol buffer or a list ofTensor
. They are most likely an output by TF summary methods liketf.summary.scalar
ortf.summary.merge_all
. It can be passed in as one tensor; if more than one, they must be passed in as a list.
Raises:
-
ValueError
: Exactly one of scaffold or summary_op should be set.
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=None)
© 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/SummarySaverHook