justheuristic 2 gadi atpakaļ
vecāks
revīzija
8b8d54abc5
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. 1 1
      src/client/remote_forward_backward.py

+ 1 - 1
src/client/remote_forward_backward.py

@@ -111,7 +111,7 @@ async def run_remote_backward(
     inputs: torch.Tensor,
     inputs: torch.Tensor,
     grad_outputs: List[torch.Tensor],
     grad_outputs: List[torch.Tensor],
     *extra_tensors: torch.Tensor,
     *extra_tensors: torch.Tensor,
-    metadata: bytes = b"",
+    **kwargs,
 ) -> Sequence[torch.Tensor]:
 ) -> Sequence[torch.Tensor]:
     """
     """
     Serializes grad outputs and calls "rpc_backward" on a remote server.
     Serializes grad outputs and calls "rpc_backward" on a remote server.