API References¶
Core API¶
LightningDataModule for loading DataLoaders with ease. |
|
Decorator for LightningModule methods. |
|
Various hooks to be used in the Lightning code. |
|
nn.Module with additional great features. |
Callbacks API¶
Abstract base class used to build new callbacks. |
|
Early Stopping |
|
GPU Stats Monitor |
|
Gradient Accumulator |
|
Learning Rate Monitor |
|
Model Checkpointing |
|
Progress Bars |
Loggers API¶
Abstract base class used to build new loggers. |
|
Comet Logger |
|
CSV logger |
|
MLflow Logger |
|
Neptune Logger |
|
TensorBoard Logger |
|
Test Tube Logger |
|
Weights and Biases Logger |
Tuner API¶
Utilities API¶
Helper functions to help with reproducibility of models. |