快捷方式

sync_sync_collector

torchrl.trainers.helpers.sync_sync_collector(env_fns: Callable | list[Callable], env_kwargs: dict | list[dict] | None, num_env_per_collector: int | None = None, num_collectors: int | None = None, **kwargs) SyncDataCollector | MultiSyncDataCollector[源码]

运行同步收集器,每个收集器运行同步环境。

例如:

../../_images/aafig-77eaa37526b10b56438e35ff9489eb12c19b4c6c.svg

环境可以是相同的,也可以是不同的。后一种情况,env_fns 应该是包含各种环境所有创建函数的列表,并且策略应批量处理这些环境。

参数:
  • env_fns – 返回 EnvBase 类实例的可调用对象(或可调用对象列表)。

  • env_kwargs – 可选。包含正在创建的环境的 kwargs 的字典(或字典列表)。

  • num_env_per_collector – 每个数据收集器的环境数量。num_env_per_collector * num_collectors 的乘积应小于或等于可用工作线程的数量。

  • num_collectors – 并行运行的数据收集器的数量。

  • **kwargs – 传递给数据收集器的其他 kwargs

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源