From 94167d6e655d255c2e665f616b9169dfd1e138eb Mon Sep 17 00:00:00 2001 From: Jin Zehao Date: Thu, 28 Mar 2024 06:16:29 +0800 Subject: [PATCH] docs: fix LightningCLI docs example for Trainer Callbacks (#19701) --- docs/source-pytorch/cli/lightning_cli_advanced_3.rst | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/source-pytorch/cli/lightning_cli_advanced_3.rst b/docs/source-pytorch/cli/lightning_cli_advanced_3.rst index ddc09dec2b..315ca5aabe 100644 --- a/docs/source-pytorch/cli/lightning_cli_advanced_3.rst +++ b/docs/source-pytorch/cli/lightning_cli_advanced_3.rst @@ -80,12 +80,12 @@ following: trainer: callbacks: - - class_path: lightning.pytorch.callbacks.EarlyStopping + - class_path: lightning.pytorch.callbacks.ModelCheckpoint init_args: - patience: 5 + save_weights_only: true - class_path: lightning.pytorch.callbacks.LearningRateMonitor init_args: - ... + logging_interval: 'epoch' Similar to the callbacks, any parameter in :class:`~lightning.pytorch.trainer.trainer.Trainer` and user extended :class:`~lightning.pytorch.core.LightningModule` and