소스 검색

restore py37

justheuristic 5 년 전
부모
커밋
2ec9f35d7e
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      hivemind/utils/autograd.py

+ 1 - 1
hivemind/utils/autograd.py

@@ -94,4 +94,4 @@ class _ParallelApplyFunction(torch.autograd.Function):
         futures = [run_in_background(run_isolated_backward, func, context, *grads)
                    for context, grads in zip(contexts, grad_outputs_per_call)]
         flat_grads_wrt_input = tuple(grad for future in futures for grad in future.result())
-        return None, None, None, None, *flat_grads_wrt_input
+        return (None, None, None, None, *flat_grads_wrt_input)