from local

This commit is contained in:
dohe0342 2023-02-03 01:23:54 +09:00
parent 914d88cd52
commit af7c1f3162
2 changed files with 1 additions and 1 deletions

View File

@ -718,7 +718,7 @@ def main():
world_size = args.world_size
assert world_size >= 1
if world_size > 1:
mp.spawn(run, args=(world_size, args), nprocs=world_size, join=True)
mp.spawn(run, args=(world_size, args, wb), nprocs=world_size, join=True)
else:
run(rank=0, world_size=1, args=args)