From 551daca04724dc82839bc70c72088706a339893e Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 18 Jul 2019 11:25:02 -0400 Subject: [PATCH] set dp as default backend --- pytorch_lightning/models/trainer.py | 1 - pytorch_lightning/pt_overrides/override_data_parallel.py | 1 + 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 2d51a3f0..1c8a18b9 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -255,7 +255,6 @@ class Trainer(TrainerIO): # ----------------- # RUN VALIDATION STEP # ----------------- - pdb.set_trace() if self.data_parallel: output = model(data_batch, batch_i) else: diff --git a/pytorch_lightning/pt_overrides/override_data_parallel.py b/pytorch_lightning/pt_overrides/override_data_parallel.py index ff4fce8c..634c4ff9 100644 --- a/pytorch_lightning/pt_overrides/override_data_parallel.py +++ b/pytorch_lightning/pt_overrides/override_data_parallel.py @@ -130,6 +130,7 @@ def parallel_apply(modules, inputs, kwargs_tup=None, devices=None): # --------------- # CHANGE + pdb.set_trace() if module.training: output = module.training_step(*input, **kwargs) else: