check_env_specs

torchrl.envs.utils.check_env_specs(env, return_contiguous=True, check_dtype=True, seed: Optional[int] = None)

针对短时间 rollout 的结果测试环境规范。

此测试函数应用作对使用 torchrl 的 EnvBase 子类封装的环境进行健全性检查：预期数据与收集的数据之间的任何差异都应引发断言错误。

损坏的环境规范可能会使并行环境无法使用。

参数:

• env (EnvBase) – 要针对数据检查规范的环境。

• return_contiguous (bool, optional) – 如果 True，则将使用 return_contiguous=True 调用随机 rollout。在某些情况下（例如，输入/输出的形状不一致）这将失败。默认为 True。

• check_dtype (bool, optional) – 如果为 False，将跳过 dtype 检查。默认为 True。

• seed (int, optional) – 为了可重复性，可以设置种子。种子将在 pytorch 中临时设置，然后 RNG 状态将恢复到之前的状态。对于环境，我们设置了种子，但由于将 rng 状态恢复到之前的状态不是大多数环境的功能，因此我们将其留给用户来完成。默认为 None。

注意：此函数会重置环境种子。它应该“离线”使用，以检查环境是否构建充分，但它可能会影响实验的种子，因此应将其排除在训练脚本之外。