From 1a39f703adc5de43f6b208fa64a8a95a7da52151 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 20 Jul 2019 09:22:04 -0400 Subject: [PATCH] removed logging --- pytorch_lightning/models/trainer.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 588e1f22..3418ae25 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -380,10 +380,7 @@ class Trainer(TrainerIO): # otherwise we launch the required number of processes try: nb_slurm_tasks = int(os.environ['SLURM_NTASKS']) - print('x'*100) - print(f'NB_TASKS: {nb_slurm_tasks}') - print('x'*100) - nb_requested_gpus = len(self.data_parallel_device_ids) + 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