快捷方式

get_stats_random_rollout

torchrl.trainers.helpers.get_stats_random_rollout(cfg: DictConfig, proof_environment: EnvBase = None, key: Optional[str] = None)[原始碼]

使用隨機 rollout 從環境中收集統計資訊(loc 和 scale)。

引數:
  • cfg (DictConfig) – 一個配置物件,包含 init_env_steps 欄位,表示收集用於計算統計資訊的總幀數。

  • proof_environment (EnvBase 例項, 可選) – 如果提供,將使用此環境執行 rollout。如果未提供,則將使用 cfg 物件建立。

  • key (str, 可選) – 如果提供,將收集此 key 的統計資訊。如果未提供,則期望 env.observation_spec 中只存在一個 key。

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深入教程

檢視教程

資源

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

檢視資源