transformed_env_constructor¶
- torchrl.trainers.helpers.transformed_env_constructor(cfg: DictConfig, video_tag: str = '', logger: Optional[Logger] = None, stats: Optional[dict] = None, norm_obs_only: bool = False, use_env_creator: bool = False, custom_env_maker: Optional[Callable] = None, custom_env: Optional[EnvBase] = None, return_transformed_envs: bool = True, action_dim_gsde: Optional[int] = None, state_dim_gsde: Optional[int] = None, batch_dims: Optional[int] = 0, obs_norm_state_dict: Optional[dict] = None) Union[Callable, EnvCreator][原始碼]¶
從使用適當的解析器建構函式構建的 argparse.Namespace 返回環境建立器。
- 引數:
cfg (DictConfig) – 包含指令碼引數的 DictConfig。
video_tag (str, optional) – 要傳遞給 Logger 物件的影片標籤
logger (Logger, optional) – 與指令碼關聯的日誌記錄器
stats (dict, optional) – 包含 ObservationNorm 轉換的
loc和scale的字典norm_obs_only (bool, optional) – 如果為 True 且使用 VecNorm，則獎勵不會被線上歸一化。預設為 False。
use_env_creator (bool, optional) – 是否應使用 EnvCreator 類。透過使用 EnvCreator，可以確保在為 VecNorm 轉換使用工作程序時，執行的統計資訊將被放入共享記憶體中，並且所有工作程序都可以訪問。預設為 True。
custom_env_maker (callable, optional) – 如果您的環境建立器不屬於 torchrl 環境包裝器，則可以改用自定義的可呼叫物件。在這種情況下，它將覆蓋從 args 中檢索到的建構函式。
custom_env (EnvBase, optional) – 如果需要轉換現有的環境，可以直接將其傳遞給此輔助函式。 custom_env_maker 和 custom_env 是互斥的功能。
return_transformed_envs (bool, optional) – 如果為
True，則返回轉換後的環境。action_dim_gsde (int, Optional) – 如果使用 gSDE，這可以表示用於初始化噪聲的操作維度。確保在並行執行的環境中指示這一點。
state_dim_gsde – 如果使用 gSDE，這可以表示用於初始化噪聲的狀態維度。確保在並行執行的環境中指示這一點。
batch_dims (int, optional) – 資料批次的維度數。如果使用單個環境，則應為 0（預設值）。如果多個環境並行轉換為，則應設定為 1（或批次的維度數）。
obs_norm_state_dict (dict, optional) – 要載入到環境中的 ObservationNorm 轉換的 state_dict
