快捷方式

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][源]

運行同步收集器,每個收集器運行同步環境。

例如:

../../_images/aafig-77eaa37526b10b56438e35ff9489eb12c19b4c6c.svg

環境可以相同或不同。如果不同,env_fns 應是包含各種環境的所有建立函式的列表,並且策略應批次處理這些環境。

引數:
  • env_fns – 可呼叫物件(或可呼叫物件列表),返回 EnvBase 類的例項。

  • env_kwargs – 可選。包含要建立的環境的關鍵字引數的字典(或字典列表)。

  • num_env_per_collector – 每個資料收集器的環境數量。num_env_per_collector * num_collectors 的乘積應小於或等於可用 workers 數量。

  • num_collectors – 並行執行的資料收集器數量。

  • **kwargs – 傳遞給資料收集器的其他關鍵字引數


© Copyright 2022, Meta。

使用 Sphinx 構建,主題由 Read the Docs 提供。

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源