Skip to content

Y
ygo-agent
Commit b1054104 authored by biluo.shen's avatar biluo.shen
Browse files
Options

Use forkserver to avoid NCCL timeout

parent a932ce02
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
ygoai/rl/dist.py
View file @ b1054104
...@@ -36,6 +36,7 @@ def setup(backend, rank, world_size, port): ...@@ -36,6 +36,7 @@ def setup(backend, rank, world_size, port):
def mp_start(run): def mp_start(run):
mp.set_start_method('forkserver')
world_size = int(os.getenv("WORLD_SIZE", "1")) world_size = int(os.getenv("WORLD_SIZE", "1"))
if world_size == 1: if world_size == 1:
run(local_rank=0, world_size=world_size) run(local_rank=0, world_size=world_size)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment