sync_sync_collector¶
- torchrl.trainers.helpers.sync_sync_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) Union[SyncDataCollector, MultiSyncDataCollector] [source]¶
运行同步收集器,每个收集器运行同步环境。
例如
环境可以相同或不同。在后一种情况下,env_fns 应该是一个列表,其中包含所有用于各种环境的创建者函数,并且策略应该以批处理方式处理这些环境。
- 参数:
env_fns – 返回 EnvBase 类实例的可调用对象(或可调用对象列表)。
env_kwargs – 可选。字典(或字典列表)包含要创建的环境的关键字参数。
num_env_per_collector – 每个数据收集器的环境数量。num_env_per_collector * num_collectors 的乘积应该小于或等于可用工作程序的数量。
num_collectors – 要并行运行的数据收集器的数量。
**kwargs – 传递给数据收集器的其他关键字参数