快捷方式

存储

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

Storage 是回放缓冲区的容器。

每个存储都必须实现 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 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源