From 8b8d54abc5255bcb9f27f742c811c2b79817f656 Mon Sep 17 00:00:00 2001 From: justheuristic Date: Sat, 10 Sep 2022 22:11:26 +0300 Subject: [PATCH] typo kwargs --- src/client/remote_forward_backward.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/client/remote_forward_backward.py b/src/client/remote_forward_backward.py index 34fa3b5..b8713ff 100644 --- a/src/client/remote_forward_backward.py +++ b/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.