diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index d84e80a7..abb4b167 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -95,7 +95,7 @@ class Trainer(TrainerIO): :param current_gpu_name: :param nb_gpu_nodes: :param gpus: - :param \: Log GPU memory utilization as metric + :param log_gpu_memory: Log GPU memory utilization as metric during training. This can lead to lower performance on some servers, in particular when `nvidia-smi` is slow. :param show_progress_bar: