tf_agents.train.triggers.ReverbCheckpointTrigger

Checkpoints data from Reverb replay buffer.

Inherits From: IntervalTrigger

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.
reverb_client the Reverb client required for checkpointing.

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.