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())
