mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-11 12:31:23 +08:00
updated args
This commit is contained in:
@@ -96,7 +96,6 @@ 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]
|
||||||
|
|||||||
Reference in New Issue
Block a user