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