Skip to content

Add option to pass args to tracker initialization #12

@SimonLarsen

Description

@SimonLarsen

Could be as simple as adding a new field tracker_kwargs that is added to init_kwargs when calling Accelerator.init_trackers.

init_kwargs={
"wandb": {"name": self.run_name},
"mlflow": {"run_name": self.run_name}
},

This solution does not work if we want to support multiple trackers at once in the future.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions