deepr.hooks.StepsPerSecHook

class deepr.hooks.StepsPerSecHook(batch_size, name=None, use_mlflow=False, use_graphite=False, skip_after_step=None, every_n_steps=100, every_n_secs=None, output_dir=None, summary_writer=None)[source]

Logs steps per seconds and num_examples_per_sec.

batch_size

Batch Size

Type:: int

prefix

Prefix of tags when sending to MLFlow / Graphite

Type:: str, Optional

use_mlflow

If True, send metrics to MLFlow. Default is False.

Type:: bool, Optional

use_graphite

If True, send metrics to Graphite. Default is False.

Type:: bool, Optional

skip_after_step

If not None, do not run the hooks after this step.

Type:: int, Optional

__init__(batch_size, name=None, use_mlflow=False, use_graphite=False, skip_after_step=None, every_n_steps=100, every_n_secs=None, output_dir=None, summary_writer=None)[source]

Methods

`__init__`(batch_size[, name, use_mlflow, ...])
`after_create_session`(session, coord)	Called when new TensorFlow session is created.
`after_run`(run_context, run_values)	Called after each call to run().
`before_run`(run_context)	Called before each call to run().
`begin`()	Called once before using the session.
`end`(session)	Called at the end of session.