tf_agents.train.triggers.StepPerSecondLogTrigger

Logs train_steps_per_second.

Inherits From: IntervalTrigger

tf_agents.train.triggers.StepPerSecondLogTrigger(
    train_step: tf.Variable, interval: int, log_to_terminal: bool = True
)

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

reset() -> None

Resets the trigger interval.

set_start

View source

set_start(
    start: int
) -> None

__call__

View source

__call__(
    value: int, force_trigger: bool = False
) -> None

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.