ray.train.lightning.LightningConfigBuilder.module

LightningConfigBuilder.module(cls: Optional[Type[pytorch_lightning.core.lightning.LightningModule]] = None, **kwargs) → ray.train.lightning.lightning_trainer.LightningConfigBuilder

Set up the Pytorch Lightning module class.

Parameters

• cls – A subclass of pytorch_lightning.LightningModule that defines your model and training logic. Note that this is a class definition instead of a class instance.

• **kwargs – The initialization argument list of your lightning module.