Struct border_async_trainer::ReplayBufferProxy

pub struct ReplayBufferProxy<R: ReplayBufferBase> { /* private fields */ }

A wrapper of replay buffer for asynchronous trainer.

Implementations

impl<R: ReplayBufferBase> ReplayBufferProxy<R>

pub fn build_with_sender(
    id: usize,
    config: &ReplayBufferProxyConfig,
    sender: Sender<PushedItemMessage<R::PushedItem>>
) -> Self

Trait Implementations

impl<R: ReplayBufferBase> ReplayBufferBase for ReplayBufferProxy<R>

type Config = ReplayBufferProxyConfig

Configuration of the replay buffer.

type PushedItem = R::PushedItem

Items pushed into the buffer.

type Batch = R::Batch

Batch generated from the buffer.

fn build(_config: &Self::Config) -> Self

Build a replay buffer from Self::Config.

fn push(&mut self, tr: Self::PushedItem) -> Result<()>

Pushes a transition into the buffer.

fn len(&self) -> usize

The number of samples in the buffer.

fn batch(&mut self, _size: usize) -> Result<Self::Batch>

Constructs a batch.

fn update_priority(
    &mut self,
    _ixs: &Option<Vec<usize>>,
    _td_err: &Option<Vec<f32>>
)

Updates priority.