Lightning-AI / Lightning-AI/pytorch-lightning
Support skipping the first iteration time in process bar
Open
Nobody has claimed this yet.
feature
help wanted
progress bar: tqdm
torch.compile
- Dominant language
- Python
- Stars
- 31.4k
- Forks
- 3.8k
- Avg merge
- 6d 7h
- Merged PRs (30d)
- 6
Description
### Description & Motivation
Because the first iteration time is incorrect (model initialization and torch.compile when first interation)
### Pitch
_No response_
### Alternatives
_No response_
### Additional context
_No response_
cc @borda @awaelchli @carmocca
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
The issue does not name a file, test, or entry point. Start by locating the process-bar implementation and its iteration-time calculation, then inspect how the first iteration includes model initialization and torch.compile. Done should mean the first iteration can be skipped when calculating the displayed process-bar timing.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning, tooling
- Issue type
- Feature
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 35/100