快捷方式

Storage

class torchrl.data.replay_buffers.Storage(max_size: int, checkpointer: StorageCheckpointerBase | None = None, compilable: bool = False)[source]

Storage 是回放缓冲区(replay buffer)的容器。

每个 Storage 都必须实现 set、get 和 __len__ 方法。Get 和 set 应支持整数以及整数列表。

Storage 不需要具有固定的大小,但如果它有固定大小,则应确保其与缓冲区大小兼容。

attach(buffer: Any) None[source]

此函数将采样器附加到此存储。

从此 Storage 读取数据的 Buffer 必须通过调用此方法作为附加实体包含进来。这可以确保当 Storage 中的数据发生更改时,组件会感知到这些更改,即使 Storage 被其他 Buffer 共享(例如,优先级采样器)。

参数:

buffer – 读取此存储的对象。

dump(*args, **kwargs)[source]

dumps() 的别名。

load(*args, **kwargs)[source]

loads() 的别名。

save(*args, **kwargs)[source]

dumps() 的别名。

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源