Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixes #280 (#309)

This commit is contained in:
William Falcon 2019-10-05 10:55:50 -04:00 committed by GitHub
parent 967957e55c
commit 2ac9f1aea7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pytorch_lightning/trainer/trainer.py
View File

@ -706,14 +706,6 @@ class Trainer(TrainerIO):
else: else:
nb_gpus = self.nb_requested_gpus nb_gpus = self.nb_requested_gpus
nb_tasks = self.nb_slurm_tasks nb_tasks = self.nb_slurm_tasks
msg = f"""
You requested {nb_gpus}s GPUs but launched {nb_tasks}s slurm tasks.
We will launch {nb_gpus}s processes for you.
We recommend you let slurm manage the processes by setting:
--ntasks-per-node={nb_gpus}s
If you're not using SLURM, ignore this message!
"""
warnings.warn(msg)
mp.spawn(self.ddp_train, nprocs=self.num_gpus, args=(model, )) mp.spawn(self.ddp_train, nprocs=self.num_gpus, args=(model, ))
# 1 gpu or dp option triggers training using DP module # 1 gpu or dp option triggers training using DP module