Skip to content

Commit a40524b

Browse files
pete-mcelroypeter.mcelroy
authored andcommitted
Decoupled checkpoint artifact path from model artifact path (Lightning-AI#20325)
Co-authored-by: peter.mcelroy <[email protected]>
1 parent f42e397 commit a40524b

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)