From dd0b6ed056d08c820e2bbac92924239a0c210497 Mon Sep 17 00:00:00 2001 From: vmoens Date: Tue, 5 Dec 2023 21:24:22 +0000 Subject: [PATCH] doc --- docs/source/reference/data.rst | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/docs/source/reference/data.rst b/docs/source/reference/data.rst index 7b8a057814a..55ebd12e867 100644 --- a/docs/source/reference/data.rst +++ b/docs/source/reference/data.rst @@ -91,7 +91,11 @@ Sharing replay buffers across processes ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ Replay buffers can be shared between processes as long as their components are -sharable. Sharable storages include :class:`~torchrl.data.replay_buffers.storages.LazyMemmapStorage` +sharable. This feature allows for multiple processes to collect data and populate a shared +replay buffer collaboratively, rather than centralizing the data on the main process +which can incur some data transmission overhead. + +Sharable storages include :class:`~torchrl.data.replay_buffers.storages.LazyMemmapStorage` or any subclass of :class:`~torchrl.data.replay_buffers.storages.TensorStorage` as long as they are instantiated and their content is stored as memory-mapped tensors. Stateful writers such as :class:`~torchrl.data.replay_buffers.writers.TensorDictRoundRobinWriter`