Function remote_method

distributed/rpc/parameter_server/rpc_parameter_server.py:79–81

(method, rref, *args, **kwargs)

# back.

def remote_method(method, rref, *args, **kwargs):
    args = [method, rref] + list(args)
    return rpc.rpc_sync(rref.owner(), call_method, args=args, kwargs=kwargs)

# --------- Parameter Server --------------------
class ParameterServer(nn.Module):