xpag.buffers.rljax_buffer.RljaxBuffer#

class RljaxBuffer(action_dim, observation_dim, buffer_size, sampler=None, gamma=0.99, nstep=1)#

Bases: Buffer

Methods

insert

Inserts a transition in the buffer

sample

Uses the sampler to returns a batch of transitions

insert(step)#

Inserts a transition in the buffer

sample(batch_size)#

Uses the sampler to returns a batch of transitions

Return type:

Dict[str, ndarray]