ray.train.lightning.RayDDPStrategy#

class ray.train.lightning.RayDDPStrategy(*args: Any, **kwargs: Any)[source]#

Bases: DDPStrategy

Subclass of DDPStrategy to ensure compatibility with Ray orchestration.

For a full list of initialization arguments, please refer to: https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.strategies.DDPStrategy.html

Note that process_group_backend, timeout, and start_method are disabled here, please specify these arguments in TorchConfig instead.

PublicAPI (beta): This API is in beta and may change before becoming stable.

Methods

Attributes

distributed_sampler_kwargs

root_device