存储
-
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() 的别名。