快捷方式

sync_async_collector

torchrl.trainers.helpers.sync_async_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) MultiaSyncDataCollector[原始碼]

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

../../_images/aafig-8621270533bb6c98244a6fe726e49583c7be9adf.svg

環境型別可以相同或不同。後一種情況,env_fns 應該是一個包含各種環境的所有建立函式的列表,並且策略應該批次處理這些環境。

引數:
  • env_fns – 返回 EnvBase 類例項的 Callable(或 Callable 列表)。

  • env_kwargs – 可選。包含正在建立的環境的 kwargs 的字典(或字典列表)。

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

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

  • **kwargs – 傳遞給資料收集器的其他 kwargs

文件

訪問全面的 PyTorch 開發者文件

檢視文件

教程

為初學者和高階開發者提供深入的教程

檢視教程

資源

查詢開發資源並讓您的問題得到解答

檢視資源