Skip to content

Commit 4281b58

Browse files
craigpfeiferBorda
andauthored
making litmodels message clearer (#20707)
* making litmodels message clearer * linter --------- Co-authored-by: Jirka B <j.borovec+github@gmail.com>
1 parent cd785a4 commit 4281b58

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/lightning/pytorch/trainer/connectors/callback_connector.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,8 +106,9 @@ def _configure_checkpoint_callbacks(self, enable_checkpointing: bool) -> None:
106106
model_checkpoint = LitModelCheckpoint(model_name=self.trainer._model_registry)
107107
else:
108108
rank_zero_info(
109-
"You are using the plain ModelCheckpoint callback."
110-
" Consider using LitModelCheckpoint which with seamless uploading to Model registry."
109+
"You are using the default ModelCheckpoint callback."
110+
" Install `litmodels` package to use the `LitModelCheckpoint` instead"
111+
" for seamless uploading to the Lightning model registry."
111112
)
112113
model_checkpoint = ModelCheckpoint()
113114
self.trainer.callbacks.append(model_checkpoint)

0 commit comments

Comments
 (0)