From 62252cee582383e0b638125e07aad8416015374e Mon Sep 17 00:00:00 2001 From: Thomas J Fan Date: Wed, 4 Sep 2019 17:46:56 -0400 Subject: [PATCH] STY Minor flake8 fix (#197) --- pytorch_lightning/models/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: