快捷方式

Storage

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

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

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

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

attach(buffer: Any) None[源代码]

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

从该存储读取的缓冲区必须通过调用此方法作为已附加实体包含进来。这确保了当存储中的数据发生变化时,组件能够感知到这些变化,即使该存储与其他缓冲区(例如,Priority Samplers)共享。

参数:

buffer – 读取此存储的对象。

dump(*args, **kwargs)[源代码]

dumps() 的别名。

load(*args, **kwargs)[源代码]

loads() 的别名。

save(*args, **kwargs)[源代码]

dumps() 的别名。

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源