From 54e9a5e1c6ca316576a2eecccb2e0c6ca4f685fe Mon Sep 17 00:00:00 2001 From: Shubham Agarwal Date: Sun, 8 Mar 2020 20:27:11 +0000 Subject: [PATCH] comment line too long --- pytorch_lightning/trainer/evaluation_loop.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/evaluation_loop.py b/pytorch_lightning/trainer/evaluation_loop.py index 071be073ba756..da0c16d0e4f52 100644 --- a/pytorch_lightning/trainer/evaluation_loop.py +++ b/pytorch_lightning/trainer/evaluation_loop.py @@ -427,7 +427,7 @@ def evaluation_forward(self, model, batch, batch_idx, dataloader_idx, test_mode: # set cuda device to root gpu # related to https://github.com/PyTorchLightning/pytorch-lightning/issues/958 - # Refer solution: https://github.com/pytorch/pytorch/issues/9871#issuecomment-408304190 + # Refer: https://github.com/pytorch/pytorch/issues/9871#issuecomment-408304190 root_device = torch.device("cuda", root_gpu) torch.cuda.set_device(root_device) else: