api_references.rst

API References

Accelerator API

pytorch_lightning.accelerators

Accelerator CPUAccelerator GPUAccelerator TPUAccelerator

Core API

pytorch_lightning.core

datamodule decorators hooks lightning

Callbacks API

pytorch_lightning.callbacks

base early_stopping gpu_stats_monitor gradient_accumulation_scheduler lr_monitor model_checkpoint progress

Loggers API

pytorch_lightning.loggers

base comet csv_logs mlflow neptune tensorboard test_tube wandb

Plugins API

Training Type Plugins

pytorch_lightning.plugins.training_type

TrainingTypePlugin SingleDevicePlugin ParallelPlugin DataParallelPlugin DDPPlugin DDP2Plugin DDPShardedPlugin DDPSpawnShardedPlugin DDPSpawnPlugin DeepSpeedPlugin HorovodPlugin RPCPlugin RPCSequentialPlugin SingleTPUPlugin TPUSpawnPlugin

Precision Plugins

pytorch_lightning.plugins.precision

PrecisionPlugin NativeMixedPrecisionPlugin ShardedNativeMixedPrecisionPlugin ApexMixedPrecisionPlugin DeepSpeedPrecisionPlugin TPUHalfPrecisionPlugin DoublePrecisionPlugin

Cluster Environments

pytorch_lightning.plugins.environments

ClusterEnvironment LightningEnvironment TorchElasticEnvironment SLURMEnvironment

Profiler API

pytorch_lightning.profiler

profilers

Trainer API

pytorch_lightning.trainer

trainer

Tuner API

pytorch_lightning.tuner.tuning

Tuner

Utilities API

pytorch_lightning.utilities

cli argparse seed