set dp as default backend

2019-07-18 11:29:38 -04:00 · 2019-07-18 11:29:38 -04:00 · 2096a0aa84
parent c253f96c53
commit 2096a0aa84
1 changed files with 1 additions and 0 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -271,6 +271,7 @@ class Trainer(TrainerIO):
            if self.use_ddp:
                output = model(data_batch, batch_i)
            elif self.use_dp:
+                pdb.set_trace()
                output = model(data_batch, batch_i)
                output = reduce_distributed_output(output, len(self.data_parallel_device_ids))