快捷方式

存储

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

存储是回放缓冲区的容器。

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

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

attach(buffer: Any) None[source]

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

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

参数:

buffer – 从该存储读取数据的对象。

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

dumps() 的别名。

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

loads() 的别名。

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

dumps() 的别名。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取初学者和高级开发人员的深入教程

查看教程

资源

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

查看资源