ray.train.lightning.RayDDPStrategy

class ray.train.lightning.RayDDPStrategy(*args: Any, **kwargs: Any)

基类：DDPStrategy

DDPStrategy 的子类，以确保与 Ray 编排的兼容性。

如需查看完整的初始化参数列表，请参阅：https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.strategies.DDPStrategy.html

注意，process_group_backend、timeout 和 start_method 在这里被禁用，请在 TorchConfig 中指定这些参数。

PublicAPI (测试版): 此API目前处于测试阶段，在成为稳定版本之前可能会发生变化。

方法

属性

distributed_sampler_kwargs
root_device