GPUStatsMonitor vs Trainer.log_gpu_memory

Is there any benefit of log_gpu_memory over GPUStatsMonitor? Why have both?

GpuUsageLogger is renamed to GPUStatsMonitor with some improvements in performance. Check the comments on this PR.

Thanks for the quick response.

But why not remove the log_gpu_memory logic and just make it automatically use a GPUStatsMonitor? The only real difference is logging the min-max which could be added to GPUStatsMonitor.

yep can be removed now I guess. create an issue??

Done!