using slurm flag to fine node nb

2019-07-08 14:01:59 -04:00 · 2019-07-08 14:01:59 -04:00 · 94da5431cd
parent 63c113d55b
commit 94da5431cd
1 changed files with 0 additions and 1 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -302,7 +302,6 @@ class Trainer(TrainerIO):

        # when GPU is called, spawn off a single worker for each gpu
        if self.on_gpu:
-            rank = 0
            self.experiment = self.experiment.get_meta_copy()
            mp.spawn(self.dp_train, nprocs=len(self.data_parallel_device_ids), args=(model, ))
        else: