336 lines
5.5 KiB
ReStructuredText
336 lines
5.5 KiB
ReStructuredText
API References
|
|
==============
|
|
|
|
.. include:: links.rst
|
|
|
|
Accelerator API
|
|
---------------
|
|
|
|
.. currentmodule:: pytorch_lightning.accelerators
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
Accelerator
|
|
CPUAccelerator
|
|
GPUAccelerator
|
|
HPUAccelerator
|
|
IPUAccelerator
|
|
TPUAccelerator
|
|
|
|
Core API
|
|
--------
|
|
|
|
.. currentmodule:: pytorch_lightning.core
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~hooks.CheckpointHooks
|
|
~hooks.DataHooks
|
|
~hooks.ModelHooks
|
|
LightningDataModule
|
|
LightningModule
|
|
~mixins.DeviceDtypeModuleMixin
|
|
~mixins.HyperparametersMixin
|
|
~optimizer.LightningOptimizer
|
|
~saving.ModelIO
|
|
|
|
Strategy API
|
|
------------
|
|
|
|
.. currentmodule:: pytorch_lightning.strategies
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
BaguaStrategy
|
|
DDP2Strategy
|
|
DDPFullyShardedStrategy
|
|
DDPShardedStrategy
|
|
DDPSpawnShardedStrategy
|
|
DDPSpawnStrategy
|
|
DDPStrategy
|
|
DataParallelStrategy
|
|
DeepSpeedStrategy
|
|
HorovodStrategy
|
|
HPUParallelStrategy
|
|
IPUStrategy
|
|
ParallelStrategy
|
|
SingleDeviceStrategy
|
|
SingleHPUStrategy
|
|
SingleTPUStrategy
|
|
Strategy
|
|
TPUSpawnStrategy
|
|
|
|
Callbacks API
|
|
-------------
|
|
|
|
.. currentmodule:: pytorch_lightning.callbacks
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
BackboneFinetuning
|
|
BaseFinetuning
|
|
BasePredictionWriter
|
|
Callback
|
|
DeviceStatsMonitor
|
|
EarlyStopping
|
|
GPUStatsMonitor
|
|
GradientAccumulationScheduler
|
|
LambdaCallback
|
|
LearningRateMonitor
|
|
ModelCheckpoint
|
|
ModelPruning
|
|
ModelSummary
|
|
ProgressBarBase
|
|
QuantizationAwareTraining
|
|
RichModelSummary
|
|
RichProgressBar
|
|
StochasticWeightAveraging
|
|
Timer
|
|
TQDMProgressBar
|
|
XLAStatsMonitor
|
|
|
|
Loggers API
|
|
-----------
|
|
|
|
.. currentmodule:: pytorch_lightning.loggers
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
|
|
base
|
|
comet
|
|
csv_logs
|
|
mlflow
|
|
neptune
|
|
tensorboard
|
|
test_tube
|
|
wandb
|
|
|
|
Loop API
|
|
--------
|
|
|
|
Base Classes
|
|
^^^^^^^^^^^^
|
|
|
|
.. currentmodule:: pytorch_lightning.loops
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dataloader.dataloader_loop.DataLoaderLoop
|
|
~base.Loop
|
|
|
|
|
|
Default Loop Implementations
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
Training
|
|
""""""""
|
|
|
|
.. currentmodule:: pytorch_lightning.loops
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~batch.TrainingBatchLoop
|
|
~epoch.TrainingEpochLoop
|
|
FitLoop
|
|
~optimization.ManualOptimization
|
|
~optimization.OptimizerLoop
|
|
|
|
|
|
Validation and Testing
|
|
""""""""""""""""""""""
|
|
|
|
.. currentmodule:: pytorch_lightning.loops
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~epoch.EvaluationEpochLoop
|
|
~dataloader.EvaluationLoop
|
|
|
|
|
|
Prediction
|
|
""""""""""
|
|
|
|
.. currentmodule:: pytorch_lightning.loops
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~epoch.PredictionEpochLoop
|
|
~dataloader.PredictionLoop
|
|
|
|
|
|
Plugins API
|
|
-----------
|
|
|
|
Precision Plugins
|
|
^^^^^^^^^^^^^^^^^
|
|
|
|
.. currentmodule:: pytorch_lightning.plugins.precision
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
ApexMixedPrecisionPlugin
|
|
DeepSpeedPrecisionPlugin
|
|
DoublePrecisionPlugin
|
|
FullyShardedNativeMixedPrecisionPlugin
|
|
HPUPrecisionPlugin
|
|
IPUPrecisionPlugin
|
|
MixedPrecisionPlugin
|
|
NativeMixedPrecisionPlugin
|
|
PrecisionPlugin
|
|
ShardedNativeMixedPrecisionPlugin
|
|
TPUBf16PrecisionPlugin
|
|
TPUPrecisionPlugin
|
|
|
|
Cluster Environments
|
|
^^^^^^^^^^^^^^^^^^^^
|
|
|
|
.. currentmodule:: pytorch_lightning.plugins.environments
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
ClusterEnvironment
|
|
KubeflowEnvironment
|
|
LightningEnvironment
|
|
LSFEnvironment
|
|
SLURMEnvironment
|
|
TorchElasticEnvironment
|
|
|
|
Checkpoint IO Plugins
|
|
^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
.. currentmodule:: pytorch_lightning.plugins.io
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
CheckpointIO
|
|
HPUCheckpointIO
|
|
TorchCheckpointIO
|
|
XLACheckpointIO
|
|
|
|
|
|
Other Plugins
|
|
^^^^^^^^^^^^^
|
|
|
|
.. currentmodule:: pytorch_lightning.plugins
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
LayerSync
|
|
NativeSyncBatchNorm
|
|
|
|
|
|
Profiler API
|
|
------------
|
|
|
|
.. currentmodule:: pytorch_lightning.profiler
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
AdvancedProfiler
|
|
PassThroughProfiler
|
|
Profiler
|
|
PyTorchProfiler
|
|
SimpleProfiler
|
|
XLAProfiler
|
|
|
|
|
|
Trainer API
|
|
-----------
|
|
|
|
.. currentmodule:: pytorch_lightning.trainer.trainer
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
Trainer
|
|
|
|
LightningLite API
|
|
-----------------
|
|
|
|
.. currentmodule:: pytorch_lightning.lite
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
|
|
LightningLite
|
|
|
|
Tuner API
|
|
---------
|
|
|
|
.. currentmodule:: pytorch_lightning.tuner.tuning
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
Tuner
|
|
|
|
Utilities API
|
|
-------------
|
|
|
|
.. currentmodule:: pytorch_lightning.utilities
|
|
|
|
.. autosummary::
|
|
:toctree: api
|
|
:nosignatures:
|
|
|
|
apply_func
|
|
argparse
|
|
cli
|
|
cloud_io
|
|
deepspeed
|
|
distributed
|
|
finite_checks
|
|
memory
|
|
model_summary
|
|
optimizer
|
|
parsing
|
|
rank_zero
|
|
seed
|
|
warnings
|