Răsfoiți Sursa

typo kwargs

justheuristic 2 ani în urmă
părinte
comite
8b8d54abc5
1 a modificat fișierele cu 1 adăugiri și 1 ștergeri
  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,
     grad_outputs: List[torch.Tensor],
     *extra_tensors: torch.Tensor,
-    metadata: bytes = b"",
+    **kwargs,
 ) -> Sequence[torch.Tensor]:
     """
     Serializes grad outputs and calls "rpc_backward" on a remote server.