diff --git a/docs/source/extensions/callbacks.rst b/docs/source/extensions/callbacks.rst index 2c9ee612ceb228..963d9934c33e70 100644 --- a/docs/source/extensions/callbacks.rst +++ b/docs/source/extensions/callbacks.rst @@ -97,6 +97,7 @@ Lightning has a few built-in callbacks. BackboneFinetuning BaseFinetuning + BasePredictionWriter Callback EarlyStopping GPUStatsMonitor @@ -105,11 +106,12 @@ Lightning has a few built-in callbacks. LearningRateMonitor ModelCheckpoint ModelPruning - BasePredictionWriter ProgressBar ProgressBarBase + RichProgressBar QuantizationAwareTraining StochasticWeightAveraging + XLAStatsMonitor ----------