快捷方式

get_stats_random_rollout

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

使用随机 rollout 从环境中收集统计信息(位置和尺度)。

参数:
  • cfg (DictConfig) – 一个配置对象,包含 init_env_steps 字段,指示要收集以计算统计信息的总帧数。

  • proof_environment (EnvBase 实例, 可选) – 如果提供,将使用此环境执行 rollout。如果未提供,将使用 cfg 对象创建。

  • key (str, 可选) – 如果提供,将收集此键的统计信息。如果未提供,则假定 env.observation_spec 中仅存在一个键。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并获得您的问题解答

查看资源