reference/generated/torchrl.data.replay_buffers.Sampler

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

Sampler

class torchrl.data.replay_buffers.Sampler(*args, **kwargs)

A generic sampler base class for composable Replay Buffers.

can_sample(storage: Storage, batch_size: int) → bool

Returns whether the sampler can draw the requested batch.