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

Update trainer.py

Browse Source
This commit is contained in:
William Falcon
2019-08-07 10:25:51 +02:00
committed by Jiri BOROVEC
parent b2d1a24999
commit 97a9a0f6c1
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
+2 -1
View File
@@ -180,7 +180,8 @@ class Trainer(TrainerIO):
# make DP and DDP mutually exclusive
# single GPU will also use DP with devices=[0]
if self.data_parallel_device_ids:
requested_gpus = self.data_parallel_device_ids is not None
if requested_gpus and len(self.data_parallel_device_ids) > 0:
self.use_dp = distributed_backend == 'dp'
self.use_ddp = distributed_backend == 'ddp'