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

Limit number of parametes logged with MLflowCallback #34276

Open
cecheta opened this issue Oct 21, 2024 · 1 comment · May be fixed by #34279
Open

Limit number of parametes logged with MLflowCallback #34276

cecheta opened this issue Oct 21, 2024 · 1 comment · May be fixed by #34279
Labels
Feature request Request for a new feature trainer

Comments

@cecheta
Copy link

cecheta commented Oct 21, 2024

Feature request

Add a new environment variable, such as MLFLOW_MAX_LOG_PARAMS, which can limit the number of parameters logged by the MLflowCallback.

Motivation

When using mlflow in Azure ML, there is a limit of 200 parameters that can be logged in a single run, meaning that when attempting to run a training job, the callback needs to be disabled entirely, or the module needs to be "monkeypatched" to limit the number of params logged.

Your contribution

I will submit a PR

@cecheta cecheta added the Feature request Request for a new feature label Oct 21, 2024
@cecheta cecheta linked a pull request Oct 21, 2024 that will close this issue
5 tasks
@LysandreJik
Copy link
Member

cc @ydshieh what do you think? :)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Feature request Request for a new feature trainer
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants