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 實例,選用) – 如果提供,此 env 將用於執行 rollout。 如果沒有,它將使用 cfg 物件建立。
key (str,選用) – 如果提供,將收集此鍵的統計資訊。 如果沒有,則預期 env.observation_spec 中僅存在一個鍵。