Skip to content

Commit 686fe02

Browse files
authored
making litmodels message clearer
1 parent cd785a4 commit 686fe02

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,8 +106,8 @@ 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 to use the LitModelCheckpoint for seamless uploading to the Lightning model registry."
111111
)
112112
model_checkpoint = ModelCheckpoint()
113113
self.trainer.callbacks.append(model_checkpoint)

0 commit comments

Comments
 (0)