torchrl.trainers.algorithms.configs.transforms.BatchSizeTransformConfig¶
- class torchrl.trainers.algorithms.configs.transforms.BatchSizeTransformConfig(batch_size: list[int] | None = None, reshape_fn: Any = None, reset_func: Any = None, env_kwarg: bool = False, _target_: str = 'torchrl.envs.transforms.transforms.BatchSizeTransform')[source]¶
Configuration for BatchSizeTransform.