Skip to content

Latest commit

 

History

History
4 lines (2 loc) · 207 Bytes

README.md

File metadata and controls

4 lines (2 loc) · 207 Bytes

GPUStatsMonitor

Pytorch lightnings DeviceStatsMonitor callback adapted the new torch.cuda.memory_stats which is overly detailed for most cases. Here is a makeshift copy of the old GPUStatsMonitor.