set_auto_unwrap_transformed_env

class torchrl.set_auto_unwrap_transformed_env(mode: bool)

一個上下文管理器或裝飾器，用於控制 TransformedEnv 是否應自動解開巢狀的 TransformedEnv 例項。

引數:

mode (bool) – 是否自動解開巢狀的 TransformedEnv 例項。如果為 False，則 TransformedEnv 將不會解開巢狀例項。預設為 True。

注意

直到 v0.9 版本，如果 TransformedEnv 被巢狀且值未顯式設定（auto_unwrap=True 預設行為）時，將引發警告。您可以透過以下方式設定 auto_unwrap_transformed_env() 的值：

• 環境變數 AUTO_UNWRAP_TRANSFORMED_ENV；

• 在指令碼開頭設定 torchrl.set_auto_unwrap_transformed_env(val: bool).set()；

• 透過將 torchrl.set_auto_unwrap_transformed_env(val: bool) 用作上下文管理器或裝飾器。

另請參閱

TransformedEnv

示例

>>> with set_auto_unwrap_transformed_env(False):
...     env = TransformedEnv(TransformedEnv(env))
...     assert not isinstance(env.base_env, TransformedEnv)
>>> @set_auto_unwrap_transformed_env(False)
... def my_function():
...     env = TransformedEnv(TransformedEnv(env))
...     assert not isinstance(env.base_env, TransformedEnv)
...     return env