check_env_specs

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

針對短暫 rollouts 的結果測試環境規範。

此測試函式應作為包裹了 torchrl 的 EnvBase 子類的環境的健全性檢查：預期資料與收集的資料之間的任何差異都應引發斷言錯誤。

損壞的環境規範很可能導致無法使用並行環境。

引數：

• env (EnvBase) – 需要根據資料檢查其規範的環境。

• return_contiguous (bool, optional) – 如果為 True，則隨機 rollout 將使用 return_contiguous=True 呼叫。這在某些情況下會失敗（例如，輸入/輸出的形狀不一致）。預設為 None（由動態規範的存在決定）。

• check_dtype (bool, optional) – 如果為 False，則跳過 dtype 檢查。預設為 True。

• seed (int, optional) – 為確保可重現性，可以設定一個種子。種子將暫時在 pytorch 中設定，然後 RNG 狀態將恢復到之前。對於環境，我們設定了種子，但由於將 RNG 狀態恢復到之前不是大多數環境的功能，因此由使用者自行完成。預設為 None。

• tensordict (TensorDict, optional) – 可選的 tensordict 例項，用於重置。

注意：此函式會重置環境種子。它應在“離線”狀態下使用，以檢查環境是否已正確構建，但它可能會影響實驗的種子設定，因此應避免在訓練指令碼中使用。