Skip to content

Commit 9e94053

Browse files
author
peter.mcelroy
committed
Decoupled checkpoint artifact path from model artifact path
1 parent 5dea36c commit 9e94053

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/pytorch/loggers/mlflow.py

+1-1
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)