瀏覽代碼

typo kwargs

justheuristic 2 年之前
父節點
當前提交
8b8d54abc5
共有 1 個文件被更改,包括 1 次插入1 次删除
  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.