快捷方式

存储

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

Storage 是回放缓冲区的容器。

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

storage 不需要具有确定的尺寸,但如果它有尺寸,则应确保其与缓冲区尺寸兼容。

attach(buffer: Any) None[source]

此函数将采样器附加到此 storage。

从此 storage 读取数据的缓冲区必须通过调用此方法作为附加实体包含在内。这保证了当 storage 中的数据发生变化时,即使 storage 与其他缓冲区(例如优先级采样器)共享,组件也能意识到这些变化。

参数:

buffer – 从此 storage 读取数据的对象。

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

dumps() 的别名。

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

loads() 的别名。

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

dumps() 的别名。


© Copyright 2022, Meta.

使用 Sphinx 构建,主题由 Read the Docs 提供。

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题解答

查看资源