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, 可選) – 如果提供,將收集此鍵的統計資料。如果未提供,則預計 env.observation_spec 中只有一個鍵。