torchrl.trainers.algorithms.configs.transforms.DTypeCastTransformConfig#
- class torchrl.trainers.algorithms.configs.transforms.DTypeCastTransformConfig(dtype: str = 'torch.float32', in_keys: list[str] | None = None, out_keys: list[str] | None = None, in_keys_inv: list[str] | None = None, out_keys_inv: list[str] | None = None, _target_: str = 'torchrl.envs.transforms.transforms.DTypeCastTransform')[source]#
Configuration for DTypeCastTransform.