typo kwargs

This commit is contained in:
justheuristic 2022-09-10 22:11:26 +03:00
parent 42cfb96965
commit 8b8d54abc5

View File

@ -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.