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
