wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-10 12:21:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

removed logging

Browse Source
This commit is contained in:
William Falcon
2019-07-20 09:22:04 -04:00
parent 955e9ea6d5
commit 1a39f703ad
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
+1 -4
View File
@@ -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