typo kwargs

pull/47/head
justheuristic 2 years ago
parent 42cfb96965
commit 8b8d54abc5

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

Loading…
Cancel
Save