Function _remote_method

distributed/rpc/rl/main.py:41–47

r""" a helper function to run method on the owner of rref and fetch back the result using RPC

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

def _remote_method(method, rref, *args, **kwargs):
    r"""
    a helper function to run method on the owner of rref and fetch back the
    result using RPC
    """
    args = [method, rref] + list(args)
    return rpc_sync(rref.owner(), _call_method, args=args, kwargs=kwargs)


class Policy(nn.Module):