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: