Decoupled checkpoint artifact path from model artifact path (#20325)

Co-authored-by: peter.mcelroy <peter.mcelroy@earthdaily.com>
This commit is contained in:
pete-mcelroy 2024-10-08 16:30:44 -07:00 committed by GitHub
parent 5dea36c5e2
commit 8ad3e29816
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -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)