Lightning-AI/pytorch-lightning
Provide a non-expert mode for the DeviceStatsMonitor
オープン
#18,652 opened on 2023/09/27
callback: device statsfeaturehelp wanted
Repository metrics
- Stars
- (26,687 個のスター)
- 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