Lightning
=========

Lightning Modules
-----------------

.. autosummary::
   :toctree: generated
   :nosignatures:

   ~mfai.pytorch.lightning_modules.clip.CLIPLightningModule
   ~mfai.pytorch.lightning_modules.gan_dgmr.DGMRLightningModule
   ~mfai.pytorch.lightning_modules.segmentation.SegmentationLightningModule

Callbacks
---------

.. autosummary::
   :toctree: generated
   :nosignatures:

   ~mfai.pytorch.callbacks.MLFlowSaveConfigCallback
   ~mfai.pytorch.callbacks.MLFlowSystemMonitorCallback
   ~mfai.pytorch.lightning_modules.clip.SaveCLIPVisualEncoderWeights