log_gpu_memory should use a GPUStatsMonitor
🐛 Bug
Trainer.log_gpu_memory should use a GPUStatsMonitor callback
The only extra feature of log_gpu_memory is the ability to print the min and max memory used. We could add that to GPUStatsMemory if deemed necessary.
cc @rohitgr7
#2932 (comment)