check_env_specs¶
- torchrl.envs.utils.check_env_specs(env, return_contiguous=True, check_dtype=True, seed: Optional[int] = None)[來源]¶
根據簡短的 rollout 結果測試環境規格。
此測試函數應作為使用 torchrl 的 EnvBase 子類別封裝的 env 的健全性檢查：預期資料與收集的資料之間的任何差異都應引發斷言錯誤。
損壞的環境規格很可能導致無法使用平行環境。
- 參數:
env (EnvBase) – 必須根據資料檢查規格的 env。
return_contiguous (bool, optional) – 如果
True，將使用 return_contiguous=True 呼叫隨機 rollout。在某些情況下（例如，輸入/輸出的異質形狀）會失敗。預設為 True。check_dtype (bool, optional) – 如果為 False，將跳過 dtype 檢查。預設為 True。
seed (int, optional) – 為了可重現性，可以設定一個 seed。該 seed 將在 pytorch 中暫時設定，然後 RNG 狀態將恢復到之前的狀態。對於 env，我們設定 seed，但由於將 rng 狀態恢復到之前狀態並不是大多數環境的功能，因此我們將其留給使用者完成。預設為
None。
注意：此函數會重設 env seed。它應該「離線」使用，以檢查 env 是否已充分構建，但它可能會影響實驗的 seed，因此應避免在訓練腳本中使用。
