Lightning-AI / Lightning-AI/pytorch-lightning
Configure WandB run name in checkpoint path via Lightning CLI
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 31.4k
- Forks
- 3.8k
- Avg merge
- 6d 7h
- Merged PRs (30d)
- 6
Description
Description & Motivation
It would be great to format the wandb run name:
wandb.run.name
into the checkpoint filename.
epoch: 10
loss: 0.5423
filename: "epoch{epoch}-loss{val_loss:.2f}" --> "epoch10-loss0.54"
filename: "{wandb.run.name}-{epoch}-{val_loss:.2f}" --> "swept_sky_54-epoch10-loss0.54"
I am not sure if this is already possible but I think the filename function in the ModelCallback can only access metrics so far.
cc @borda @carmocca @mauvilsa
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 ModelCallback filename formatting path used by Lightning CLI and inspect how checkpoint names currently access metrics. Verify whether the W&B run name is available there, then define the expected behavior for the provided template example. Done means the run name can be included without breaking existing metric-based filenames.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- cli, machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 38/100