check_env_specs

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

测试环境规范与简短回滚结果的一致性。

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

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

参数：

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

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

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

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

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