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[源代码]¶
运行同步收集器，每个收集器运行同步环境。
例如：
环境可以相同也可以不同。在后一种情况下，`env_fns` 应该是包含所有各种环境的创建函数的列表，并且策略应该批量处理这些环境。
- 参数:
env_fns – 返回 `EnvBase` 类实例的可调用对象（或可调用对象列表）。
env_kwargs – 可选。包含正在创建的环境的 kwargs 的字典（或字典列表）。
num_env_per_collector – 每个数据收集器的环境数量。`num_env_per_collector` * `num_collectors` 的乘积应小于或等于可用工作进程的数量。
num_collectors – 要并行运行的数据收集器的数量。
**kwargs – 传递给数据收集器的其他 kwargs
