torchrl.trainers.algorithms.configs.envs_libs.BraxEnvConfig¶
- class torchrl.trainers.algorithms.configs.envs_libs.BraxEnvConfig(_partial_: bool = False, env_name: str = '???', categorical_action_encoding: bool = False, cache_clear_frequency: int | None = None, from_pixels: bool = False, frame_skip: int | None = None, device: str = 'cpu', batch_size: list[int] | None = None, allow_done_after_reset: bool = False, requires_grad: bool = False, _target_: str = 'torchrl.envs.libs.brax.BraxEnv')[source]¶
Configuration for BraxEnv environment.
