Skip to content

Commit

Permalink
Decoupled checkpoint artifact path from model artifact path (#20325)
Browse files Browse the repository at this point in the history
Co-authored-by: peter.mcelroy <peter.mcelroy@earthdaily.com>
  • Loading branch information
pete-mcelroy and peter.mcelroy authored Oct 8, 2024
1 parent 5dea36c commit 8ad3e29
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/lightning/pytorch/loggers/mlflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -360,7 +360,7 @@ def _scan_and_log_checkpoints(self, checkpoint_callback: ModelCheckpoint) -> Non
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)
Expand Down

0 comments on commit 8ad3e29

Please sign in to comment.