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)[原始碼]¶
在 Dreamer 中記錄解碼觀測值的轉換。
示例
>>> model_based_env = DreamerEnv(...) >>> model_based_env_eval = model_based_env.append_transform(DreamerDecoder())
