From d3b621dfd2f4a8d4a2faef04c9f78582da87477a Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 25 Jun 2019 20:04:27 -0400 Subject: [PATCH] updated args --- pytorch_lightning/pt_overrides/override_data_parallel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/pt_overrides/override_data_parallel.py b/pytorch_lightning/pt_overrides/override_data_parallel.py index 19470106d7..1076c8a639 100644 --- a/pytorch_lightning/pt_overrides/override_data_parallel.py +++ b/pytorch_lightning/pt_overrides/override_data_parallel.py @@ -83,7 +83,6 @@ def parallel_apply(modules, inputs, kwargs_tup=None, devices=None): with lock: results[i] = e - pdb.set_trace() if len(modules) > 1: threads = [threading.Thread(target=_worker, args=(i, module, input, kwargs, device)) @@ -97,6 +96,7 @@ def parallel_apply(modules, inputs, kwargs_tup=None, devices=None): else: _worker(0, modules[0], inputs[0], kwargs_tup[0], devices[0]) + pdb.set_trace() outputs = [] for i in range(len(inputs)): output = results[i]