ray.train.lightning.LightningConfigBuilder.strategy
ray.train.lightning.LightningConfigBuilder.strategy#
- LightningConfigBuilder.strategy(name: str = 'ddp', **kwargs) ray.train.lightning.lightning_trainer.LightningConfigBuilder[source]#
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