SA: for Lightning-AI#958: removing root gpu hack in trainer/evaluatio…

…n_loop
shubhamagarwal92 · Mar 15, 2020 · dc31ab9 · dc31ab9
1 parent cc34c94
commit dc31ab9
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/pytorch_lightning/trainer/evaluation_loop.py b/pytorch_lightning/trainer/evaluation_loop.py
@@ -424,7 +424,6 @@ def evaluation_forward(self, model, batch, batch_idx, dataloader_idx, test_mode:
 
             if isinstance(self.data_parallel_device_ids, list):
                 root_gpu = self.data_parallel_device_ids[0]
-
                 # set cuda device to root gpu
                 # related to https://github.com/PyTorchLightning/pytorch-lightning/issues/958
                 # Refer: https://github.com/pytorch/pytorch/issues/9871#issuecomment-408304190