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 [原始碼]¶
執行非同步收集器,每個收集器執行同步環境。
環境類型可以是相同的或不同的。在後一種情況下,env_fns 應該是一個列表,其中包含各種環境的所有建立器函數,並且策略應該批量處理這些環境。
- 參數:
env_fns – Callable(或 Callables 的列表),傳回 EnvBase 類別的實例。
env_kwargs – 可選。包含正在建立的環境之 kwargs 的字典(或字典的列表)。
num_env_per_collector – 每個資料收集器的環境數量。乘積 num_env_per_collector * num_collectors 應小於或等於可用的 worker 數量。
num_collectors – 要並行執行的資料收集器數量。
**kwargs – 傳遞給資料收集器的其他 kwargs