DreamerDecoder¶
- torchrl.envs.model_based.dreamer.DreamerDecoder(in_keys: Sequence[NestedKey] = None, out_keys: Sequence[NestedKey] | None = None, in_keys_inv: Sequence[NestedKey] | None = None, out_keys_inv: Sequence[NestedKey] | None = None)[source]¶
用于在 Dreamer 中记录解码观察结果的转换。
示例
>>> model_based_env = DreamerEnv(...) >>> model_based_env_eval = model_based_env.append_transform(DreamerDecoder())
