Lightning¶

Lightning Modules¶

`CLIPLightningModule`
`DGMRLightningModule`	Pytorch Lightning Module to train the GAN.
`SegmentationLightningModule`

`MLFlowSaveConfigCallback`	A Lightning callback to save the config.yaml in the run directory instead of in the top-level save_dir.
`MLFlowSystemMonitorCallback`	A Lightning callback to log system metrics (GPU usage etc.) in MLFlow.
`SaveCLIPVisualEncoderWeights`	Callback to save the weights of the visual encoder during training.