ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf_agents.train.triggers.StepPerSecondLogTrigger

Logs train_steps_per_second.

Inherits From: IntervalTrigger

Used in the notebooks

Used in the tutorials

train_step tf.Variable which keeps track of the number of train steps.
interval How often, in train_steps, the trigger will save. Note that as long as the >= interval number of steps have passed since the last trigger, the event gets triggered. The current value is not necessarily interval steps away from the last triggered value.
log_to_terminal If True ouputs steps/sec to logging.info.

Methods

reset

View source

Resets the trigger interval.

set_start

View source

__call__

View source

Maybe trigger the event based on the interval.

Args
value the value for triggering.
force_trigger If True, the trigger will be forced triggered unless the last trigger value is equal to value.