comment line too long

Lightning-AI · Mar 8, 2020 · 54e9a5e · 54e9a5e
1 parent 6d89505
commit 54e9a5e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: