Add MLFlow run name option in config (#1961)

Co-authored-by: Adam Hazell <adam.hazell@mindfoundry.ai>
This commit is contained in:
Adam Hazell
2024-10-11 18:33:06 +01:00
committed by GitHub
parent e73b8dff8d
commit 922db77521
3 changed files with 8 additions and 3 deletions

View File

@@ -266,6 +266,7 @@ wandb_log_model: # "checkpoint" to log model to wandb Artifacts every `save_step
# mlflow configuration if you're using it
mlflow_tracking_uri: # URI to mlflow
mlflow_experiment_name: # Your experiment name
mlflow_run_name: # Your run name
hf_mlflow_log_artifacts: # set to true to copy each saved checkpoint on each save to mlflow artifact registry
# Comet configuration if you're using it