Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

updated args

This commit is contained in:
William Falcon 2019-06-25 20:04:27 -04:00
parent ac88e3f832
commit d3b621dfd2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/pt_overrides/override_data_parallel.py
View File

@ -83,7 +83,6 @@ def parallel_apply(modules, inputs, kwargs_tup=None, devices=None):
with lock: with lock:
results[i] = e results[i] = e
pdb.set_trace()
if len(modules) > 1: if len(modules) > 1:
threads = [threading.Thread(target=_worker, threads = [threading.Thread(target=_worker,
args=(i, module, input, kwargs, device)) args=(i, module, input, kwargs, device))
@ -97,6 +96,7 @@ def parallel_apply(modules, inputs, kwargs_tup=None, devices=None):
else: else:
_worker(0, modules[0], inputs[0], kwargs_tup[0], devices[0]) _worker(0, modules[0], inputs[0], kwargs_tup[0], devices[0])
pdb.set_trace()
outputs = [] outputs = []
for i in range(len(inputs)): for i in range(len(inputs)):
output = results[i] output = results[i]