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 Goto Github PK
View Code? Open in Web Editor NEWLicense: Apache License 2.0