Lightning-AI/pytorch-lightning
GitHub で見るProvide a non-expert mode for the DeviceStatsMonitor
Open
#18,652 opened on 2023年9月27日
callback: device statsfeaturehelp wanted
Repository metrics
- Stars
- (26,687 stars)
- PR merge metrics
- (平均マージ 9d 15h) (30d で 3 merged PRs)
説明
Description & Motivation
The DeviceStatsMonitor currently dumps the entire torch.cuda.memory_stats into the logger. Beginner users might be overwhelmed by the amount of data logged and are confused which metrics matter to maximize their utilization.
Pitch
Make the callback more user-friendly by logging only core metrics like GPU utilization and total memory usage. Provide an expert mode that when enabled logs what we see today.
Alternatives
No response
Additional context
This was proposed by user magical_arjay on our Discord.
cc @borda @awaelchli