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。
