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.
cemde/GPUStatsMonitor
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|