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