ray.train.lightning.prepare_trainer

ray.train.lightning.prepare_trainer#

ray.train.lightning.prepare_trainer(trainer: pytorch_lightning.Trainer) → pytorch_lightning.Trainer[source]#

Prepare the PyTorch Lightning Trainer for distributed execution.

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