justheuristic 5 lat temu
rodzic
commit
4b3eda1aa6
1 zmienionych plików z 1 dodań i 1 usunięć
  1. 1 1
      tesseract/utils/autograd.py

+ 1 - 1
tesseract/utils/autograd.py

@@ -79,7 +79,7 @@ class _ParallelApplyFunction(torch.autograd.Function):
 
         futures = [run_in_background(run_isolated_forward, func, *args) for args in args_per_call]
 
-        outputs, contexts = zip(*[future.result() for future in futures])
+        contexts, outputs = zip(*[future.result() for future in futures])
         output_strides = np.cumsum([0] + list(map(len, outputs)))
         ctx._inner_func = func
         ctx._call_contexts = contexts