log_gpu_memory should use a GPUStatsMonitor #3333
Labels
feature
Is an improvement or enhancement
good first issue
Good for newcomers
help wanted
Open to be worked on
refactor
won't fix
This will not be worked on
🐛 Bug
Trainer.log_gpu_memory
should use aGPUStatsMonitor
callbackThe only extra feature of
log_gpu_memory
is the ability to print the min and max memory used. We could add that toGPUStatsMemory
if deemed necessary.cc @rohitgr7
#2932 (comment)
The text was updated successfully, but these errors were encountered: