Lightning-AI / Lightning-AI/litgpt
Add support for providing `wandb` run name.
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 13.7k
- Forks
- 1.5k
- Avg merge
- 15h 37m
- Merged PRs (30d)
- 1
Description
At the moment, the logger functionality for `wandb` only sets the project name (based on the provided model). It would be useful to also allow us to set the run's name, as well as the group:
```
if logger_name == "wandb":
run_name=kwargs.pop("run_name", os.environ.get("WANDB_RUN_NAME"))
group=kwargs.pop("group", os.environ.get("WANDB_GROUP_NAME"))
return WandbLogger(name=run_name, group=group, project=name, resume=resume, **kwargs)
```
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start at the logger functionality that handles logger_name="wandb" and inspect how the project currently derives the project name from the model. Trace the WandbLogger construction and its kwargs handling. Done means callers can provide run_name and group, with the documented environment-variable fallbacks, without changing existing project or resume behavior.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 2/5
- Estimated time
- 1-3 hours
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 45/100