Skip to content

Commit 8ad3e29

Browse files
pete-mcelroypeter.mcelroy
andauthored
Decoupled checkpoint artifact path from model artifact path (#20325)
Co-authored-by: peter.mcelroy <peter.mcelroy@earthdaily.com>
1 parent 5dea36c commit 8ad3e29

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/pytorch/loggers/mlflow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -360,7 +360,7 @@ def _scan_and_log_checkpoints(self, checkpoint_callback: ModelCheckpoint) -> Non
360360
aliases = ["latest", "best"] if p == checkpoint_callback.best_model_path else ["latest"]
361361

362362
# Artifact path on mlflow
363-
artifact_path = f"model/checkpoints/{Path(p).stem}"
363+
artifact_path = Path(p).stem
364364

365365
# Log the checkpoint
366366
self.experiment.log_artifact(self._run_id, p, artifact_path)

0 commit comments

Comments
 (0)