ray.train.lightning.LightningConfigBuilder.strategy

LightningConfigBuilder.strategy(name: str = 'ddp', **kwargs) → ray.train.lightning.lightning_trainer.LightningConfigBuilder

Set up the configurations of pytorch_lightning.strategies.Strategy.

Parameters

• name – The name of your distributed strategy. You can choose from “ddp”, “fsdp”, and “deepspeed”. Default: “ddp”.

• kwargs – For valid arguments to pass, please refer to: https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.strategies.DDPStrategy.html , https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.strategies.FSDPStrategy.html and https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.strategies.DeepSpeedStrategy.html