快捷方式

sync_async_collector

torchrl.trainers.helpers.sync_async_collector(env_fns: Union[Callable, List[Callable]], env_kwargs: Optional[Union[dict, List[dict]]], num_env_per_collector: Optional[int] = None, num_collectors: Optional[int] = None, **kwargs) MultiaSyncDataCollector[source]

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

../../_images/aafig-8621270533bb6c98244a6fe726e49583c7be9adf.svg

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

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

  • env_kwargs – 可选。字典(或字典列表),包含正在创建的环境的关键字参数。

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

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

  • **kwargs – 传递给数据收集器的其他关键字参数

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源