API References¶

Core API¶

`datamodule`	LightningDataModule for loading DataLoaders with ease.
`decorators`	Decorator for LightningModule methods.
`hooks`	Various hooks to be used in the Lightning code.
`lightning`	nn.Module with additional great features.

`base`	Abstract base class used to build new callbacks.
`early_stopping`	Early Stopping
`gpu_stats_monitor`	GPU Stats Monitor
`gradient_accumulation_scheduler`	Gradient Accumulator
`lr_monitor`	Learning Rate Monitor
`model_checkpoint`	Model Checkpointing
`progress`	Progress Bars

`base`	Abstract base class used to build new loggers.
`comet`	Comet Logger
`csv_logs`	CSV logger
`mlflow`	MLflow Logger
`neptune`	Neptune Logger
`tensorboard`	TensorBoard Logger
`test_tube`	Test Tube Logger
`wandb`	Weights and Biases Logger

Profiler to check if there are any bottlenecks in your code.

Trainer to automate the training.

`batch_size_scaling`
`lr_finder`

`argparse_utils`
`seed`	Helper functions to help with reproducibility of models.