View source on GitHub |
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 |
---|
Args | |
---|---|
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
reset() -> None
Resets the trigger interval.
set_start
set_start(
start: int
) -> None
__call__
__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 .
|