typo kwargs

priority-tasks
justheuristic 2 years ago
parent 42cfb96965
commit 8b8d54abc5

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

Loading…
Cancel
Save