Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add MLFlow run name option in config #1961

Merged

Conversation

awhazell
Copy link
Contributor

@awhazell awhazell commented Oct 11, 2024

Description

Adds an mlflow_run_name option to the config. Currently exclusive with the wandb name, however this matches current functionality where the run_name is set to None if wandb is not in use

#1959

Motivation and Context

We are using axolotl with mlflow to do many fine tuning runs on varying datasets and parameters, it would be nice to be able to tag them when generating the configs rather than having to query the parameters. e.g. 'dataset-a-learning-rate-2e-5'

How has this been tested?

  • Running an unchanged config to check no breakages
  • Running a new config with an additional mlflow_run_name parameter in the config and checking that the run name is set as desired

Screenshots (if appropriate)

Types of changes

Social Handles (Optional)

@winglian winglian merged commit 922db77 into axolotl-ai-cloud:main Oct 11, 2024
9 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants