GPU memory show max/min only
Instead of logging GPU mem for all GPUs, probably better to just show the min/max usage? Maybe the flag can be modified?
# now
Trainer(log_gpu_memory=False)  

# proposed
Trainer(log_gpu_memory='min_max')  

# and for all GPUs
Trainer(log_gpu_memory='all')  
@adefazio