From 8ad3e29816a63d8ce5c00ac104b14729a4176f4f Mon Sep 17 00:00:00 2001 From: pete-mcelroy <147437841+pete-mcelroy@users.noreply.github.com> Date: Tue, 8 Oct 2024 16:30:44 -0700 Subject: [PATCH] Decoupled checkpoint artifact path from model artifact path (#20325) Co-authored-by: peter.mcelroy --- src/lightning/pytorch/loggers/mlflow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lightning/pytorch/loggers/mlflow.py b/src/lightning/pytorch/loggers/mlflow.py index 4437d5f8a7..1b15014cd0 100644 --- a/src/lightning/pytorch/loggers/mlflow.py +++ b/src/lightning/pytorch/loggers/mlflow.py @@ -360,7 +360,7 @@ class MLFlowLogger(Logger): aliases = ["latest", "best"] if p == checkpoint_callback.best_model_path else ["latest"] # Artifact path on mlflow - artifact_path = f"model/checkpoints/{Path(p).stem}" + artifact_path = Path(p).stem # Log the checkpoint self.experiment.log_artifact(self._run_id, p, artifact_path)