GPUStatsMonitor vs Trainer.log_gpu_memory

callbacks
1

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

2

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

3

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.

4

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

5

Done!

https://github.com/PyTorchLightning/pytorch-lightning/issues/3333