ray.train.lightning.LightningConfigBuilder.trainer#

LightningConfigBuilder.trainer(**kwargs) → ray.train.lightning.lightning_trainer.LightningConfigBuilder[source]#

Set up the configurations of pytorch_lightning.Trainer.

Note that you don’t have to specify the strategy, device and num_nodes arguments here, since the LightningTrainer creates a PyTorch Lightning Strategy object with the configurations specified in the strategy() method. The device and num_nodes are also configured automatically by the LightningTrainer. If no configuration is specified, it creates a DDPStrategy by default.

For accelerator, currently only "cpu" and "gpu" are supported.

Parameters

kwargs – The initialization arguments for pytorch_lightning.Trainer For valid arguments to pass, please refer to: https://lightning.ai/docs/pytorch/stable/common/trainer.html#init.