ray.rllib.evaluation.rollout_worker.RolloutWorker.setup_torch_data_parallel

RolloutWorker.setup_torch_data_parallel(url: str, world_rank: int, world_size: int, backend: str) → None

Join a torch process group for distributed SGD.