Lightning-AI/pytorch-lightning

Support skipping the first iteration time in process bar

Open

#19,357 建立於 2024年1月28日

在 GitHub 查看
 (1 留言) (1 反應) (0 負責人)Python (26,687 star) (3,233 fork)batch import
featurehelp wantedprogress bar: tqdmtorch.compile

描述

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

貢獻者指南

技術棧
python
領域
machine learning
議題類型
feature
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
half day
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
stale
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
clear
前置要求
PythonPyTorch Lightning basics
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
50
研究方向
Look at the progress bar implementation in pytorch lightning, likely in pytorch lightning/callbacks/progress bar.py. Identify where the iteration time is computed and displayed. The first iteration time is incorrect due to model initialization and torch.compile overhead. Consider adding an option to skip the first iteration time or adjust its calculation. Check if there are any related comments or existing patterns in the codebase.