From 7e053fc73156f7c2a25764b86606e974e19ab327 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sun, 21 Jul 2019 12:18:46 -0400 Subject: [PATCH] added analysis notebook --- 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 91de1b69..ba5e7272 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -393,9 +393,9 @@ class Trainer(TrainerIO): # whenever we have the correct number of tasks, we let slurm manage processes # otherwise we launch the required number of processes + nb_requested_gpus = len(self.data_parallel_device_ids) * self.nb_gpu_nodes try: nb_slurm_tasks = int(os.environ['SLURM_NTASKS']) - nb_requested_gpus = len(self.data_parallel_device_ids) * self.nb_gpu_nodes is_slurm_managing_tasks = nb_slurm_tasks == nb_requested_gpus except Exception as e: # likely not on slurm, so set the slurm managed flag to false