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