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

cleaning up demos

Browse Source
This commit is contained in:
William Falcon
2019-10-05 15:57:31 -04:00
parent 170262ff73
commit a3df04f049
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/trainer.py
+2 -2
View File
@@ -536,7 +536,7 @@ class Trainer(TrainerIO):
args.append(dataloader_idx)
# handle DP, DDP forward
if self.use_ddp or self.use_dp:
if self.use_ddp or self.use_dp or self.use_ddp2:
output = model(*args)
return output
@@ -1217,7 +1217,7 @@ class Trainer(TrainerIO):
if len(self.optimizers) > 1:
args.append(opt_idx)
if self.use_ddp:
if self.use_ddp or self.use_ddp2:
output = self.model(*args)
elif self.use_dp:
output = self.model(*args)